Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ import os
|
|
7 |
# Download ImageNet labels
|
8 |
os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt")
|
9 |
|
10 |
-
model = torch.hub.load('
|
11 |
# or any of these variants
|
12 |
# model = torch.hub.load('PingoLH/Pytorch-HarDNet', 'hardnet85', pretrained=True)
|
13 |
# model = torch.hub.load('PingoLH/Pytorch-HarDNet', 'hardnet68ds', pretrained=True)
|
7 |
# Download ImageNet labels
|
8 |
os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt")
|
9 |
|
10 |
+
model = torch.hub.load('AK391/Pytorch-HarDNet', 'hardnet68', pretrained=True)
|
11 |
# or any of these variants
|
12 |
# model = torch.hub.load('PingoLH/Pytorch-HarDNet', 'hardnet85', pretrained=True)
|
13 |
# model = torch.hub.load('PingoLH/Pytorch-HarDNet', 'hardnet68ds', pretrained=True)
|