Spaces:

chansung
/

segformer-tf-transformers

Runtime error

chansung commited on Jul 25, 2022

Commit

7e04741

1 Parent(s): 8eb51c3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -165,12 +165,11 @@ def ade_palette():
         [92, 0, 255],
     ]
-url='https://github.com/deep-diver/segformer-tf-transformers/releases/download/1.0/segformer-b5-finetuned-ade-640-640.onnx'
 labels_list = []
 colormap = np.asarray(ade_palette())
-model_path = wget.download(url)
-sess = ort.InferenceSession(model_path)
 csv.field_size_limit(sys.maxsize)

         [92, 0, 255],
     ]
 labels_list = []
 colormap = np.asarray(ade_palette())
+model_filename = 'segformer-b5-finetuned-ade-640-640.onnx'
+sess = ort.InferenceSession(model_filename)
 csv.field_size_limit(sys.maxsize)