Update app.py
Browse files
app.py
CHANGED
@@ -39,9 +39,9 @@ mx.test_utils.download('https://s3.amazonaws.com/onnx-model-zoo/synset.txt')
|
|
39 |
with open('synset.txt', 'r') as f:
|
40 |
labels = [l.rstrip() for l in f]
|
41 |
|
42 |
-
os.system("wget https://github.com/
|
43 |
|
44 |
-
ort_session = ort.InferenceSession("
|
45 |
|
46 |
|
47 |
def predict(path):
|
|
|
39 |
with open('synset.txt', 'r') as f:
|
40 |
labels = [l.rstrip() for l in f]
|
41 |
|
42 |
+
os.system("wget https://github.com/onnx/models/raw/main/vision/classification/inception_and_googlenet/googlenet/model/googlenet-9.onnx")
|
43 |
|
44 |
+
ort_session = ort.InferenceSession("googlenet-9.onnx")
|
45 |
|
46 |
|
47 |
def predict(path):
|