Update app.py

app.py

@@ -12,7 +12,7 @@ def predict(image):
     model = models.resnet50(pretrained=True)
     num_ftrs = model.fc.in_features
     model.fc = nn.Linear(num_ftrs, 1)
-    model.load_state_dict(torch.load("model.pth", map_location=torch.device('cpu')))
+    model.load_state_dict(torch.load("best_f1.pth", map_location=torch.device('cpu')))
     model.eval()
     
     # Preprocess image