Update app.py

app.py

@@ -69,8 +69,10 @@ def main_infer(image_input_path, prompt):
         pred_mask = pred_masks
         os.makedirs(temp_dir, exist_ok=True)
         seg_result = visualize(pred_mask, image_input_path, temp_dir)
-    
-    return answer, seg_result
+        return answer, seg_result
+    else:
+        return answer, None
+
 
 # Gradio UI