chansung commited on
Commit
7e04741
1 Parent(s): 8eb51c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -165,12 +165,11 @@ def ade_palette():
165
  [92, 0, 255],
166
  ]
167
 
168
- url='https://github.com/deep-diver/segformer-tf-transformers/releases/download/1.0/segformer-b5-finetuned-ade-640-640.onnx'
169
  labels_list = []
170
  colormap = np.asarray(ade_palette())
171
 
172
- model_path = wget.download(url)
173
- sess = ort.InferenceSession(model_path)
174
 
175
  csv.field_size_limit(sys.maxsize)
176
 
 
165
  [92, 0, 255],
166
  ]
167
 
 
168
  labels_list = []
169
  colormap = np.asarray(ade_palette())
170
 
171
+ model_filename = 'segformer-b5-finetuned-ade-640-640.onnx'
172
+ sess = ort.InferenceSession(model_filename)
173
 
174
  csv.field_size_limit(sys.maxsize)
175