Update app.py

app.py

@@ -46,10 +46,10 @@ def classify_image(inp):
     label_order = ["Burger King", "KFC", "McDonalds", "Other", "Starbucks", "Subway"]
 
     # Assuming prediction is a dictionary with label keys
-    result = [f"{label}: {prediction[label]:.2f}" for label in label_order]
-    return ", ".join(result)
-    # result = [f"{label}: {prediction:.2f}" for label, prediction in labels.items()]
+    # result = [f"{label}: {prediction[label]:.2f}" for label in label_order]
     # return ", ".join(result)
+    result = [f"{label}: {prediction[y]:.2f}" for label, y in labels.items()]
+    return ", ".join(result)