Source code for medium story 🚀Serve a TensorFlow 2 Deep Learning Image Classifier as a REST API and Dockerize it! (with GPU support)

make_test_inference.py

test_image =  read_image("../data/raw/test/82_automobile.png", image_size) 
plt.imshow(test_image)
plt.title('test image')
plt.show()

pred_softmax = model.predict(test_image.reshape(1, image_size[0], image_size[1], 3))
pred_class_index = np.argmax(pred_softmax, axis=1)

target_names = ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']
print("CNN sees", target_names[pred_class_index.item()])