Spaces:

Jfink09
/

EfficientFundus

Runtime error

Jfink09 commited on Mar 3, 2023

Commit

6dd9a3f

1 Parent(s): 40a931d

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ efficientnet_b0, efficientnet_b0_transforms = create_efficientnet_b0_model(
 # Load saved weights
 efficientnet_b0.load_state_dict(
     torch.load(
-        f="pretrained_efficientnet_b0_feature_extractor_drappcompressed.pth",
         map_location=torch.device("cpu"),  # load to CPU
     )
 )

 # Load saved weights
 efficientnet_b0.load_state_dict(
     torch.load(
+        f="pretrained_efficientnet_b0.pth",
         map_location=torch.device("cpu"),  # load to CPU
     )
 )