Update app.py
Browse files
app.py
CHANGED
|
@@ -6,8 +6,8 @@ import os
|
|
| 6 |
|
| 7 |
torch.hub.download_url_to_file("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg")
|
| 8 |
|
| 9 |
-
|
| 10 |
-
|
| 11 |
|
| 12 |
os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt")
|
| 13 |
|
|
|
|
| 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.9.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")
|
| 13 |
|