Update app.py

app.py

@@ -65,7 +65,7 @@ def main_infer(image_input_path, prompt):
     
     if '[SEG]' in answer and Visualizer is not None:
         temp_dir = tempfile.mkdtemp()
-        pred_mask = pred_masks[0]
+        pred_mask = pred_masks
         os.makedirs(temp_dir, exist_ok=True)
         seg_result = visualize(pred_mask, image_input_path, temp_dir)