Update app.py

app.py

@@ -37,8 +37,8 @@ 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)
+    segment = pipeline("image-segmentation", model=f"{model}")
+    pred = segment(image)
     pred_object_list = [str(i)+'_'+x['label'] for i, x in enumerate(pred)]
     seg_box=[]
     for i in range(len(pred)):