Hugging Face's logo

Spaces:
coderx7
/
simplenet
Sleeping

App Files Community
coderx7 commited on
Commit
62ac5e3
·
1 Parent(s): a40de8c

use m1

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ 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: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)
 
8
 
9
 
10
  import torch
11
+ model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_5m_m1", 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)