Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
173 |
-
sess = ort.InferenceSession(
|
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 |
|