Update app.py

app.py

@@ -17,7 +17,11 @@ NUM_CLASSES = len(classes)
 
 # Load model
 model = FoodIngredientClassifier(NUM_CLASSES)
-checkpoint = torch.load("best_model.pth", map_location=DEVICE)
+checkpoint = torch.load(
+    "best_model.pth",
+    map_location=DEVICE,
+    weights_only=False
+)
 model.load_state_dict(checkpoint["model_state_dict"])
 model.to(DEVICE)
 model.eval()