Hugging Face's logo

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

App Files Community
jiuface commited on
Commit
c304f92
•
1 Parent(s): b46a036

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -66,10 +66,10 @@ def get_segmentation_pipeline(
66
  Tuple[AutoImageProcessor, UperNetForSemanticSegmentation]: segmentation pipeline
67
  """
68
  image_processor = AutoImageProcessor.from_pretrained(
69
- "openmmlab/upernet-convnext-small"
70
  )
71
  image_segmentor = UperNetForSemanticSegmentation.from_pretrained(
72
- "openmmlab/upernet-convnext-small"
73
  )
74
  return image_processor, image_segmentor
75
 
 
66
  Tuple[AutoImageProcessor, UperNetForSemanticSegmentation]: segmentation pipeline
67
  """
68
  image_processor = AutoImageProcessor.from_pretrained(
69
+ "openmmlab/upernet-convnext-xlarge"
70
  )
71
  image_segmentor = UperNetForSemanticSegmentation.from_pretrained(
72
+ "openmmlab/upernet-convnext-xlarge"
73
  )
74
  return image_processor, image_segmentor
75