akhaliq HF Staff commited on
Commit
20e3a92
·
1 Parent(s): 5fae62b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,13 +12,13 @@ os.system("wget https://s3.amazonaws.com/onnx-model-zoo/synset.txt")
12
  with open('synset.txt', 'r') as f:
13
  labels = [l.rstrip() for l in f]
14
 
15
- os.system("wget https://github.com/onnx/models/raw/main/vision/classification/shufflenet/model/shufflenet-v2-12-int8.onnx")
16
 
17
  os.system("wget https://s3.amazonaws.com/model-server/inputs/kitten.jpg")
18
 
19
 
20
 
21
- model_path = 'shufflenet-v2-12-int8.onnx'
22
  model = onnx.load(model_path)
23
  session = ort.InferenceSession(model.SerializeToString())
24
 
 
12
  with open('synset.txt', 'r') as f:
13
  labels = [l.rstrip() for l in f]
14
 
15
+ os.system("wget https://github.com/AK391/models/raw/main/vision/classification/shufflenet/model/shufflenet-v2-10.onnx")
16
 
17
  os.system("wget https://s3.amazonaws.com/model-server/inputs/kitten.jpg")
18
 
19
 
20
 
21
+ model_path = 'shufflenet-v2-10.onnx'
22
  model = onnx.load(model_path)
23
  session = ort.InferenceSession(model.SerializeToString())
24