akhaliq HF Staff commited on
Commit
25d3510
·
1 Parent(s): f56da98

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/AK391/models/raw/main/vision/classification/alexnet/model/bvlcalexnet-7.onnx")
43
 
44
- ort_session = ort.InferenceSession("bvlcalexnet-7.onnx")
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):