ChestXrayClassification

Sleeping

hamsteryang commited on Aug 30, 2023

Commit

99fe99c

1 Parent(s): d89a42d

change model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -55,7 +55,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model(
 # Load saved weights
 effnetb2.load_state_dict(
     torch.load(
-        f="alexnet_pretrained.pth",
         map_location=torch.device("cpu"),  # load to CPU
     )
 )

 # Load saved weights
 effnetb2.load_state_dict(
     torch.load(
+        f="FL_global_model.pt",
         map_location=torch.device("cpu"),  # load to CPU
     )
 )