Update app.py

app.py

@@ -137,7 +137,7 @@ model = models.resnet34(pretrained=False)  # Load base ResNet-34
 model.fc = torch.nn.Linear(model.fc.in_features, 2)  # Modify final layer
 
 # Load weights properly
-checkpoint = torch.load("./resnet34.pkl", map_location="cpu")
+checkpoint = torch.load("resnet34.pth", map_location="cpu")
 
 # Extract just the model weights (ignore optimizer states)
 if "network" in checkpoint:  # Handle different checkpoint formats