Update app.py

app.py

@@ -41,7 +41,7 @@ def image_vision(image_input_path, prompt):
 def main_infer(image_input_path, prompt):
 
     answer, seg_image = image_vision(image_input_path, prompt)
-    return answer, seg_image
+    return answer, seg_image[0]
 
 # Gradio UI