Hugging Face's logo

Spaces:
jinfengxie
/
BFM_segmentation
Running

App Files Community
jinfengxie commited on
Commit
1fe39b3
·
verified ·
1 Parent(s): d544165

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -121,7 +121,7 @@ id2label= {1: '动物皮',
121
  65: '纱窗',
122
  0: '背景/未知'}
123
 
124
- model_save_path =r"D:\pythonwork\python_model\DMS_v1\model_result\model0820-100epoch-material_seg-b3"
125
  model = TFSegformerForSemanticSegmentation.from_pretrained(model_save_path)
126
 
127
  def predict_and_visualize(image):
 
121
  65: '纱窗',
122
  0: '背景/未知'}
123
 
124
+ model_save_path =“infengxie/BFM_segformer0821”
125
  model = TFSegformerForSemanticSegmentation.from_pretrained(model_save_path)
126
 
127
  def predict_and_visualize(image):