Hugging Face's logo

Spaces:
MohammadAliMKH
/
food101_classification
Build error

App Files Community
MohammadAliMKH commited on
Commit
ac12ce6
•
1 Parent(s): 192dae6

Update model.py

Browse files

add map_location=torch.device('cpu')

Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -6,4 +6,4 @@ efficient_weight = EfficientNet_B2_Weights.DEFAULT
6
 
7
  efficient_transformer = efficient_weight.transforms()
8
 
9
- efficient_model = torch.load("efficient_model_101.pth").to("cpu")
 
6
 
7
  efficient_transformer = efficient_weight.transforms()
8
 
9
+ efficient_model = torch.load("efficient_model_101.pth" , map_location=torch.device('cpu'))