Update app.py

app.py

@@ -116,7 +116,7 @@ def load_model():
         logger.info(f"Loading model from {config.MODEL_PATH}")
         model = PlantDiseaseModel(num_classes=len(config.CLASS_NAMES), dropout=0.4)
         
-        checkpoint = torch.load(config.MODEL_PATH, map_location=config.DEVICE)
+        checkpoint = torch.load(config.MODEL_PATH, map_location=config.DEVICE,weights_only=True)
         model.load_state_dict(checkpoint['model_state_dict'])
         model.to(config.DEVICE)
         model.eval()