Hugging Face's logo

Spaces:
coderx7
/
simplenet
Sleeping

App Files Community
coderx7 commited on
Commit
30539f3
·
verified ·
1 Parent(s): 3f08774

add trust_repo=True

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from torchvision import transforms
7
 
8
  os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt")
9
 
10
- model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_5m_m1", pretrained=True)
11
  # or any of these variants
12
  # model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_5m_m2", pretrained=True)
13
  # model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_9m_m1", pretrained=True)
 
7
 
8
  os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt")
9
 
10
+ model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_5m_m1", pretrained=True, trust_repo=True)
11
  # or any of these variants
12
  # model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_5m_m2", pretrained=True)
13
  # model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_9m_m1", pretrained=True)