Update app.py

app.py

@@ -4,20 +4,15 @@ from transformers import pipeline
 import gradio as gr
 import os
 
-models1 = [
+models = [
     "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]
+current_model = models[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")
 
@@ -42,6 +37,7 @@ def image_objects(image):
 def get_seg(image, model_choice):
     image = img_resize(image)
     model = models[model_choice]
+    prediction = pipeline("image-segmentation", model=f"{model}")
     pred = model(image)
     pred_object_list = [str(i)+'_'+x['label'] for i, x in enumerate(pred)]
     seg_box=[]