Update app.py

app.py

@@ -40,7 +40,7 @@ def classify_image(inp):
 
     label_order = ["Burger King", "KFC", "McDonalds", "Other", "Starbucks", "Subway"]
 
-    result = {label: int(f"{prediction[0][labels[label]]:.6f}") for label in label_order}
+    result = {label: float(f"{prediction[0][labels[label]]:.6f}") for label in label_order}
 
     return result