rgera commited on
Commit
d4c0cb7
·
1 Parent(s): 63d49d2

update app.py forgot .pth after model name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ effnetb2_2, effnetb2_transforms_2 = create_effnetb2() # we dont need "model.crae
20
  # load the saved weigths
21
  effnetb2_2.load_state_dict(
22
  torch.load(
23
- f = "09_pretrained_effnetb2_feature_extractor_20_percent",
24
  map_location= torch.device("cpu")
25
  )
26
  )
 
20
  # load the saved weigths
21
  effnetb2_2.load_state_dict(
22
  torch.load(
23
+ f = "09_pretrained_effnetb2_feature_extractor_20_percent.pth",
24
  map_location= torch.device("cpu")
25
  )
26
  )