AmanullaChishty commited on
Commit
12b72bb
·
verified ·
1 Parent(s): 3c7bee7

Updates model name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model(
20
  # Load saved weights
21
  effnetb2.load_state_dict(
22
  torch.load(
23
- f="09_pretrained_effnetb2_feature_extractor_pizza_steak_sushi_20_percent.pth",
24
  map_location=torch.device("cpu"), # load to CPU
25
  )
26
  )
 
20
  # Load saved weights
21
  effnetb2.load_state_dict(
22
  torch.load(
23
+ f="09_effnetb2_feature_extractor.pth",
24
  map_location=torch.device("cpu"), # load to CPU
25
  )
26
  )