Update app.py

app.py

@@ -61,9 +61,10 @@ def image_vision(image_input_path, prompt):
 def main_infer(image_input_path, prompt):
 
     answer, seg_image = image_vision(image_input_path, prompt)
-    pred_masks = seg_image[0]
+    
     
     if '[SEG]' in answer and Visualizer is not None:
+        pred_masks = seg_image[0]
         temp_dir = tempfile.mkdtemp()
         pred_mask = pred_masks
         os.makedirs(temp_dir, exist_ok=True)