Spaces:

onnx
/

Inception_v2

Runtime error

akhaliq HF Staff commited on Mar 10, 2022

Commit

25cc91f

1 Parent(s): ccc8c53

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -39,9 +39,9 @@ mx.test_utils.download('https://s3.amazonaws.com/onnx-model-zoo/synset.txt')
 with open('synset.txt', 'r') as f:
     labels = [l.rstrip() for l in f]
-os.system("wget https://github.com/onnx/models/raw/main/vision/classification/inception_and_googlenet/inception_v1/model/inception-v1-12.onnx")
-ort_session = ort.InferenceSession("inception-v1-12.onnx")
 def predict(path):

 with open('synset.txt', 'r') as f:
     labels = [l.rstrip() for l in f]
+os.system("wget https://github.com/AK391/models/raw/main/vision/classification/inception_and_googlenet/inception_v2/model/inception-v2-9.onnx")
+ort_session = ort.InferenceSession("inception-v2-9.onnx")
 def predict(path):