Hugging Face's logo

Spaces:
Aedelon
/
foodvision_big
Sleeping

App Files Community
Aedelon commited on
Commit
6ee4675
•
1 Parent(s): 7f0d284

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -18,8 +18,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model(
18
 
19
  # Load saved weights
20
  effnetb2.load_state_dict(
21
- torch.load(f="pretrained_effnetb2_food101_20_percent.pth"),
22
- map_location=torch.device("cpu")
23
  )
24
 
25
  # Prediction
 
18
 
19
  # Load saved weights
20
  effnetb2.load_state_dict(
21
+ torch.load(f="pretrained_effnetb2_food101_20_percent.pth", map_location=torch.device("cpu"))
 
22
  )
23
 
24
  # Prediction