Update app.py

app.py

@@ -7,7 +7,22 @@ from transformers import pipeline
 import gradio as gr
 import os
 
-model = pipeline("image-segmentation", model="facebook/detr-resnet-50-panoptic")
+models1 = [
+    "facebook/detr-resnet-50-panoptic",
+    "CIDAS/clipseg-rd64-refined",
+    "facebook/maskformer-swin-large-ade",
+    "nvidia/segformer-b1-finetuned-cityscapes-1024-1024",    
+]
+current_model = models1[0]
+
+models=[
+    pipeline("image-segmentation", model="facebook/detr-resnet-50-panoptic"),
+    pipeline("image-segmentation", model="facebook/maskformer-swin-large-ade"),
+    pipeline("image-segmentation", model="CIDAS/clipseg-rd64-refined"),
+    pipeline("image-segmentation", model="nvidia/segformer-b1-finetuned-cityscapes-1024-1024"),
+]
+
+#model = pipeline("image-segmentation", model="facebook/detr-resnet-50-panoptic")
 
 pred = []
 
@@ -24,9 +39,12 @@ def image_objects(image):
     pred = model(image)
     pred_object_list = [str(i)+'_'+x['label'] for i, x in enumerate(pred)]
     return gr.Dropdown.update(choices = pred_object_list, interactive = True)
+
+
     
-def get_seg(image):
+def get_seg(image, model_choice):
     image = img_resize(image)
+    model = models[model_choice]
     pred = model(image)
     pred_object_list = [str(i)+'_'+x['label'] for i, x in enumerate(pred)]
     seg_box=[]
@@ -60,17 +78,16 @@ with app:
             """)
     with gr.Row():
         with gr.Column():
-         
             image_input = gr.Image(label="Input Image",type="pil")
-    with gr.Row():
+            model_name = gr.Dropdown(show_label=False, choices=[m for m in models], type="index", value=current_model, interactive=True)
         with gr.Column():
-                object_output = gr.Dropdown(label="Objects")
-    
+            gal1=gr.Gallery(type="filepath").style(grid=6)
     with gr.Row():
         with gr.Column():
-            gal1=gr.Gallery(type="filepath").style(grid=4)
+                object_output = gr.Dropdown(label="Objects")
 
-    image_input.change(get_seg, inputs=[image_input], outputs=[gal1,object_output])
+    
+    image_input.change(get_seg, inputs=[image_input, model_name], outputs=[gal1,object_output])