Update app.py

app.py

@@ -7,7 +7,7 @@ import os
 import numpy as np
 
 
-labels= {'Burger King': 0, 'KFC': 1,'McDonalds': 2,'Other': 3,'Starbucks': 4,'Subway': 5}
+# labels= {'Burger King': 0, 'KFC': 1,'McDonalds': 2,'Other': 3,'Starbucks': 4,'Subway': 5}
 HEIGHT,WIDTH=224,224
 NUM_CLASSES=6
 
@@ -43,7 +43,7 @@ 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]}")
-    return f"{prediction}...{prediction[0][2]}..{predicted_class_indices}" 
+    return f"{labels[i]: float(prediction[i]) for i in range(NUM_CLASSES)}"