Update app.py

app.py

@@ -108,7 +108,7 @@ def load_model():
     model.fc = nn.Linear(num_ftrs, 12)  # Adjust to the number of classes you have
     
     # Load the state dict
-    model.load_state_dict(torch.load('resnet50_garbage_classificationv1.2.pth', map_location=torch.device('cpu')))
+    model.load_state_dict(torch.load('resnet50_garbage_classification.pth', map_location=torch.device('cpu')))
     
     model.eval()  # Set to evaluation mode
     return model