change model

2023-08-31 14:06

app.py

@@ -36,7 +36,7 @@ def create_densenet121_model(num_classes: int = 2, seed: int = 42):
 densenet, densenet_transforms = create_densenet121_model()
 
 # Load saved weights
-state_dict = torch.load("FL_global_model.pt", map_location=torch.device("cpu"))
+state_dict = torch.load("model/FL_global_model_4be885f7-8d33-4498-a5ef-85aa301706bd.pt", map_location=torch.device("cpu"))
 model_weights = state_dict["model"]
 densenet.load_state_dict(model_weights,strict=False)  # Set strict to True since we now expect it to match