coderx7 commited on
Commit
a40de8c
·
1 Parent(s): 956819a

use the explicit v1 form

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -8,15 +8,15 @@ os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_cl
8
 
9
 
10
  import torch
11
- model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_5m_m2", pretrained=True)
12
  # or any of these variants
13
- # model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_5m_m2", pretrained=True)
14
- # model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_9m_m1", pretrained=True)
15
- # model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_9m_m2", pretrained=True)
16
- # model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_small_m1_05", pretrained=True)
17
- # model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_small_m2_05", pretrained=True)
18
- # model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_small_m1_075", pretrained=True)
19
- # model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_small_m2_075", pretrained=True)
20
  model.eval()
21
 
22
  # Download an example image from the pytorch website
 
8
 
9
 
10
  import torch
11
+ model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_5m_m2", pretrained=True)
12
  # or any of these variants
13
+ # model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_5m_m2", pretrained=True)
14
+ # model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_9m_m1", pretrained=True)
15
+ # model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_9m_m2", pretrained=True)
16
+ # model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_small_m1_05", pretrained=True)
17
+ # model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_small_m2_05", pretrained=True)
18
+ # model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_small_m1_075", pretrained=True)
19
+ # model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_small_m2_075", pretrained=True)
20
  model.eval()
21
 
22
  # Download an example image from the pytorch website