Update app.py

app.py

@@ -12,7 +12,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # Define the model architecture
 model = models.resnet152()
 model.fc = torch.nn.Linear(model.fc.in_features, 26)  # Adjust for the number of classes
-model.load_state_dict(torch.load("model.pth", map_location=device))
+model.load_state_dict(torch.load("trained_model.pth", map_location=device))
 model = model.to(device)
 model.eval()