danyalmalik commited on
Commit
a7394d1
·
1 Parent(s): b727665

fixed typos

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -49,10 +49,10 @@ class Net(nn.Module):
49
  net = Net()
50
  net.to(device)
51
 
52
- model = huggingface_hub.hf_hub_download(
53
- 'danyalmalik/sceneryclassifier', '1655322438.4798265_Acc0.693_modelweights.pth')
54
 
55
- net.load_state_dict(torch.load(os.path.join('/models/', model)))
56
 
57
  labels = ['Buildings', 'Forest', 'Glacier', 'Mountain', 'Sea', 'Street']
58
 
 
49
  net = Net()
50
  net.to(device)
51
 
52
+ model = huggingface_hub.cached_download(huggingface_hub.hf_hub_url(
53
+ 'danyalmalik/sceneryclassifier', '1655322438.4798265_Acc0.693_modelweights.pth'))
54
 
55
+ net.load_state_dict(torch.load(model))
56
 
57
  labels = ['Buildings', 'Forest', 'Glacier', 'Mountain', 'Sea', 'Street']
58