Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -121,7 +121,7 @@ id2label= {1: '动物皮',
|
|
121 |
65: '纱窗',
|
122 |
0: '背景/未知'}
|
123 |
|
124 |
-
model_save_path
|
125 |
model = TFSegformerForSemanticSegmentation.from_pretrained(model_save_path)
|
126 |
|
127 |
def predict_and_visualize(image):
|
|
|
121 |
65: '纱窗',
|
122 |
0: '背景/未知'}
|
123 |
|
124 |
+
model_save_path ='jinfengxie/BFM_segformer0821'
|
125 |
model = TFSegformerForSemanticSegmentation.from_pretrained(model_save_path)
|
126 |
|
127 |
def predict_and_visualize(image):
|