Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -13,11 +13,11 @@ mx.test_utils.download('https://s3.amazonaws.com/onnx-model-zoo/synset.txt')
|
|
13 |
with open('synset.txt', 'r') as f:
|
14 |
labels = [l.rstrip() for l in f]
|
15 |
|
16 |
-
os.system("wget https://github.com/
|
17 |
|
18 |
# Enter path to the ONNX model file
|
19 |
|
20 |
-
sym, arg_params, aux_params = import_model('
|
21 |
|
22 |
Batch = namedtuple('Batch', ['data'])
|
23 |
def get_image(path, show=False):
|
|
|
13 |
with open('synset.txt', 'r') as f:
|
14 |
labels = [l.rstrip() for l in f]
|
15 |
|
16 |
+
os.system("wget https://github.com/AK391/models/blob/main/vision/classification/shufflenet/model/shufflenet-v2-10.onnx")
|
17 |
|
18 |
# Enter path to the ONNX model file
|
19 |
|
20 |
+
sym, arg_params, aux_params = import_model('shufflenet-v2-10.onnx')
|
21 |
|
22 |
Batch = namedtuple('Batch', ['data'])
|
23 |
def get_image(path, show=False):
|