Update app.py

app.py

@@ -33,7 +33,12 @@ class FoodIngredientClassifier(nn.Module):
 # -----------------------------
 # Load checkpoint
 # -----------------------------
-checkpoint = torch.load("best_model.pth", map_location=device)
+checkpoint = torch.load(
+    "best_model.pth",
+    map_location=device,
+    weights_only=False
+)
+
 
 mlb = checkpoint["mlb"]
 class_names = mlb.classes_