Update app.py

app.py

@@ -26,7 +26,7 @@ def predict_dr(image):
     try:
         # Load model (you'll need to update this path)
         model = create_model()
-        model.load_state_dict(torch.load('model.pth', map_location=torch.device('cpu')))
+        model.load_state_dict(torch.load('diabeticRetinopathy.pth', map_location=torch.device('cpu')))
         model.eval()
         
         # Preprocess image