tested done

Signed-off-by: Suvaditya Mukherjee <[email protected]>

app.py

@@ -4,9 +4,6 @@ from tensorflow import keras
 from PIL import Image
 import os
 
-# To force inference using CPU only
-os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
-
 # Model definition
 
 image_input = keras.Input(shape=(None, None, 3))
@@ -34,12 +31,12 @@ model = keras.Model(inputs=image_input, outputs=model_output)
 
 
 def inference(image: tf.Tensor):
-    tensor = keras.preprocessing.image.img_to_array(image)
-    tensor = tf.expand_dims(tensor, axis=0)
+    image = tf.expand_dims(image, axis=0)
     y = model(image).numpy()
     preds = keras.applications.imagenet_utils.decode_predictions(y, top=5)
     result = {i[1]: str(i[2]) for i in preds[0]}
     result = {k: v for k, v in sorted(result.items(), key=lambda item: item[1])}
+    result = sorted(result.items(), key=lambda x: x[1], reverse=True)
     return result
 
 def greet(name):