ai-model-002

Running on Zero

jiuface commited on Aug 2

Commit

facc283

•

1 Parent(s): f52bca8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -247,10 +247,10 @@ def get_segmentation_pipeline(
         Tuple[AutoImageProcessor, UperNetForSemanticSegmentation]: segmentation pipeline
     """
     image_processor = AutoImageProcessor.from_pretrained(
-        "openmmlab/upernet-convnext-xlarge"
     )
     image_segmentor = UperNetForSemanticSegmentation.from_pretrained(
-        "openmmlab/upernet-convnext-xlarge"
     )
     return image_processor, image_segmentor

         Tuple[AutoImageProcessor, UperNetForSemanticSegmentation]: segmentation pipeline
     """
     image_processor = AutoImageProcessor.from_pretrained(
+        "openmmlab/upernet-convnext-small"
     )
     image_segmentor = UperNetForSemanticSegmentation.from_pretrained(
+        "openmmlab/upernet-convnext-small"
     )
     return image_processor, image_segmentor