Update app.py

app.py

@@ -56,12 +56,12 @@ def predict_class(image):
 
 def classify_image(image):
     results = predict_class(image)
-    output = {}
+    # output = {}
     # for index in range(len(results)):
     #     predicted_label = labels.get(index)
     #     score = results[index]
     #     output[predicted_label] = str(score)
-    output = {labels.get(i): float(prediction[i]) for i in range(len(results))}
+    output = {labels.get(i): float(results[i]) for i in range(len(results))}
     return output