added table in output

app.py

@@ -37,8 +37,6 @@ def read_image(image):
     print(image.shape)
 #     plt.imshow(image)
 #     plt.show()
-    # image = tf.image.resize(images=image, size=[IMG_SIZE, IMG_SIZE])
-    # image = image / 127.5 - 1
     image, _ = preprocess_image(image, 1) #  1 here is a temporary label
     return image
 
@@ -47,11 +45,9 @@ def infer(input_image):
     image_tensor = read_image(input_image)
     print(image_tensor.shape)
     predictions = model.predict(np.expand_dims((image_tensor), axis=0))
-    predictions = np.squeeze(predictions)
-    predictions = np.argmax(predictions) # , axis=2
-    predicted_label = class_names[predictions.item()]
-    return str(predicted_label)
+    predictions = np.squeeze(predictions).astype(float)
     
+    return dict(zip(class_names, predictions))
     
 # get the inputs
 input = gr.inputs.Image(shape=(IMG_SIZE, IMG_SIZE))