akhaliq HF staff commited on
Commit
a95f60e
1 Parent(s): 4ffdb9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/onnx/models/raw/main/vision/classification/alexnet/model/bvlcalexnet-12.onnx")
17
 
18
  # Enter path to the ONNX model file
19
 
20
- sym, arg_params, aux_params = import_model('bvlcalexnet-12.onnx')
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/raw/main/vision/classification/alexnet/model/bvlcalexnet-7.onnx")
17
 
18
  # Enter path to the ONNX model file
19
 
20
+ sym, arg_params, aux_params = import_model('bvlcalexnet-7.onnx')
21
 
22
  Batch = namedtuple('Batch', ['data'])
23
  def get_image(path, show=False):