Hugging Face's logo

Spaces:
jiuface
/
ai-model-002
Running on Zero

App Files Community
jiuface commited on
Commit
facc283
·
verified ·
1 Parent(s): f52bca8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -247,10 +247,10 @@ def get_segmentation_pipeline(
247
  Tuple[AutoImageProcessor, UperNetForSemanticSegmentation]: segmentation pipeline
248
  """
249
  image_processor = AutoImageProcessor.from_pretrained(
250
- "openmmlab/upernet-convnext-xlarge"
251
  )
252
  image_segmentor = UperNetForSemanticSegmentation.from_pretrained(
253
- "openmmlab/upernet-convnext-xlarge"
254
  )
255
  return image_processor, image_segmentor
256
 
 
247
  Tuple[AutoImageProcessor, UperNetForSemanticSegmentation]: segmentation pipeline
248
  """
249
  image_processor = AutoImageProcessor.from_pretrained(
250
+ "openmmlab/upernet-convnext-small"
251
  )
252
  image_segmentor = UperNetForSemanticSegmentation.from_pretrained(
253
+ "openmmlab/upernet-convnext-small"
254
  )
255
  return image_processor, image_segmentor
256