Update app.py

app.py

@@ -57,10 +57,11 @@ def predict_class(image):
 def classify_image(image):
     results = predict_class(image)
     output = {}
-    for index in range(len(results)):
-        predicted_label = labels.get(index)
-        score = results[index]
-        output[predicted_label] = str(score)
+    # 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))}
     return output