Spaces:

Nigus
/

APP1

Sleeping

Nigus commited on May 8, 2022

Commit

d1d4d0e

1 Parent(s): 127e53b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import os
 torch.hub.download_url_to_file("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg")
-model = torch.hub.load('pytorch/vision:v0.10.0', 'shufflenetv2_x2_0', pretrained=False)
 model.load_state_dict(torch.load('shufflenetv2_x2_0-8be3c8ee.pth'))
 os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt")

 torch.hub.download_url_to_file("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg")
+model = torch.hub.load('pytorch/vision:v0.10.0', 'shufflenet_v2_x2_0', pretrained=False)
 model.load_state_dict(torch.load('shufflenetv2_x2_0-8be3c8ee.pth'))
 os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt")