Spaces:

HangenYuu
/

SeeFood101v1

Runtime error

HangenYuu commited on Jun 24, 2023

Commit

458d932

1 Parent(s): 54da27f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 import torchvision.transforms as transforms
 inference_model = timm.create_model('swin_large_patch4_window7_224', pretrained=False, num_classes=101)
-inference_model.load_state_dict(torch.load('model.pth'), map_location=torch.device('cpu'))
 inference_model.eval()
 with open('labels.txt', 'r') as f:

 import torchvision.transforms as transforms
 inference_model = timm.create_model('swin_large_patch4_window7_224', pretrained=False, num_classes=101)
+inference_model.load_state_dict(torch.load('model.pth', map_location=torch.device('cpu')))
 inference_model.eval()
 with open('labels.txt', 'r') as f: