Update app.py

app.py

@@ -43,8 +43,13 @@ def classify_image(inp):
     #             result[label] = float(predicted_class_indices[i])
     #         except KeyError:
     #             print(f"KeyError: Label not found for index {predicted_class_indices[i]}")
-    result = [f"{label}: {prediction:.2f}" for label, prediction in labels.items()]
+    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()]
+    # return ", ".join(result)