shukurullo2004 commited on
Commit
058bf4f
1 Parent(s): c7fc65b

Update app.py

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