Hugging Face's logo

Spaces:
pytorch
/
HardNet
Runtime error

App Files Community
akhaliq HF Staff commited on
Commit
00af62a
·
1 Parent(s): d190fc5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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('PingoLH/Pytorch-HarDNet', 'hardnet68', pretrained=True,device='cpu')
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)