Hugging Face's logo

Spaces:
Jfink09
/
DeepFundus
Running

App Files Community
1
Jfink09 commited on
Commit
45b564d
•
1 Parent(s): 3718d60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ resnet50, resnet50_transforms = create_resnet50_model(
28
  )
29
  resnet50.fc = nn.Linear(2048, 10)
30
 
31
- # Load saved weights
32
  resnet50.load_state_dict(
33
  torch.load(
34
  f="pretrained_resnet50_feature_extractor_drappcompressed.pth",
 
28
  )
29
  resnet50.fc = nn.Linear(2048, 10)
30
 
31
+ # Load the saved weights
32
  resnet50.load_state_dict(
33
  torch.load(
34
  f="pretrained_resnet50_feature_extractor_drappcompressed.pth",