Update app.py

app.py

@@ -12,7 +12,7 @@ model.fc = nn.Linear(num_features, num_classes)
 
 # Load the trained model weights
 try:
-    model.load_state_dict(torch.load("test 3 .pt", map_location=torch.device('cpu')))
+    model.load_state_dict(torch.load("best_model.pth", map_location=torch.device('cpu')))
     print("Model loaded successfully.")
 except Exception as e:
     print(f"Error loading model: {e}")