Update app.py

app.py

@@ -46,7 +46,6 @@ def predict_class(image):
     img = tf.image.resize(img, [input_shape[0], input_shape[1]])
     img = tf.expand_dims(img, axis=0)
     prediction = model.predict(img)
-    print(prediction)
     return prediction
 
 # UI Design
@@ -58,9 +57,9 @@ def predict_class(image):
 def classify_image(image):
     results = predict_class(image)
     output = {}
-    for prediction in results:
-        predicted_label = prediction['label']
-        score = prediction['score']
+    for _, prediction in enumerate(results):
+        predicted_label = labels[_]
+        score = prediction[_]
         output[predicted_label] = score
     return output