Update app.py

app.py

@@ -10,7 +10,7 @@ load_model = tf.keras.models.load_model('./resnet50_food_model')
 def predict(img):
     img = img.reshape(-1,224,224,3)
     prediction= load_model.predict(img)[0]
-
+    labels = ["Food", "Non-food"]
     return {labels[i]: float(prediction[i]) for i in range(2)}
 
 title = "Food Classifier"