Update app.py

app.py

@@ -37,17 +37,15 @@ def classify_image(inp):
     inp = tf.keras.applications.nasnet.preprocess_input(inp)
     prediction = model.predict(inp)
     predicted_class_indices = np.argmax(prediction, axis=1)
-    result = {}
 
     label_order = ["Burger King", "KFC", "McDonalds", "Other", "Starbucks", "Subway"]
 
-    # Create a dictionary to hold the results
-    result = {label: f"{prediction[labels[label]]:.2f}" for label in label_order}
+    result = {label: f"{prediction[0][labels[label]]:.2f}" for label in label_order}
+
+    return result
+
 
-    # Convert the dictionary to a string
-    result_str = ", ".join([f"{label}: {value}" for label, value in result.items()])
 
-    return result_str