Update app.py

app.py

@@ -30,6 +30,8 @@ model=load_model('best_model.h5')
 
 def classify_image(inp):
     np.random.seed(143)
+    labels = {'Burger King': 0, 'KFC': 1, 'McDonalds': 2, 'Other': 3, 'Starbucks': 4, 'Subway': 5}
+    NUM_CLASSES = 6
     inp = inp.reshape((-1, HEIGHT, WIDTH, 3))
     inp = tf.keras.applications.nasnet.preprocess_input(inp)
     prediction = model.predict(inp)
@@ -42,7 +44,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 prediction    
+    return result