changed model in Segmentation pipeline

Author: Vishnunkumar

Diff for: clipcrop/clipcrop.py

@@ -141,7 +141,7 @@ def __init__(self, input_path, input_text):
 
   def load_models(self):
 
-    segmentor = pipeline("image-segmentation")
+    segmentor = pipeline("image-segmentation", model="facebook/maskformer-swin-tiny-coco")
     model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
     processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32") 
 
@@ -191,4 +191,4 @@ def segment_image(self, segmentor, model, processor, num=None):
       seg_dict["score"] = scores[x]
       seg_list.append(seg_dict)
 
-    return seg_list
+    return seg_list