Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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/
|
16 |
|
17 |
os.system("wget https://s3.amazonaws.com/model-server/inputs/kitten.jpg")
|
18 |
|
19 |
|
20 |
|
21 |
-
model_path = 'shufflenet-v2-
|
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 |
|