update app.py 2023-08-31 08:05

app.py

@@ -35,7 +35,7 @@ def create_densenet121_model(num_classes: int = 1, seed: int = 42):
     return model, transforms
 
 # Create densenet121 model
-densenet, densenet_transforms = create_densenet121_model(num_classes=2)
+densenet, densenet_transforms = create_densenet121_model(num_classes=1)
 
 # Load saved weights
 # densenet.load_state_dict(torch.load("FL_global_model.pt", map_location=torch.device("cpu")))