Hugging Face's logo

Spaces:
Nigus
/
APP1
Sleeping

App Files Community
Nigus commited on
Commit
23b6452
·
1 Parent(s): 6aa74ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import os
6
 
7
  torch.hub.download_url_to_file("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg")
8
 
9
- model = torch.hub.load('pytorch/vision:v0.10.0', 'shufflenet_v2_x1_0', pretrained=False)
10
  model.load_state_dict(torch.load('shufflenetv2_x1-5666bf0f80.pth'))
11
 
12
  os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt")
 
6
 
7
  torch.hub.download_url_to_file("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg")
8
 
9
+ model = torch.hub.load('pytorch/vision:v0.10.0', 'shufflenet_v2_x1_0', pretrained=True)
10
  model.load_state_dict(torch.load('shufflenetv2_x1-5666bf0f80.pth'))
11
 
12
  os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt")