Update app.py

app.py

@@ -41,7 +41,7 @@ DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 THRESHOLD = 0.5
 
 def load_model():
-    checkpoint = torch.load("model.pth", map_location=DEVICE)
+    checkpoint = torch.load("model.pth", map_location=DEVICE, weights_only=False)
     mlb = checkpoint["mlb"]
     num_classes = len(mlb.classes_)
     model = FoodIngredientClassifier(num_classes)