Update app.py

app.py

@@ -224,7 +224,7 @@ def cellpose_segment(img_pil, resize = 1000):
     b1 = gr.DownloadButton(visible=True, value = fname_masks)
     b2 = gr.DownloadButton(visible=True, value = fname_out) #"outlines.png")
     
-    return outpix, overlay, flows, b1, b2
+    return outpix, flows, b1, b2
 
 # Gradio Interface
 #iface = gr.Interface(
@@ -307,7 +307,7 @@ with gr.Blocks(title = "Hello",
     
     #input_image.change(update_image, input_image, [input_image, filepath])
     #up_btn.upload(upload_file, up_btn, [input_image, up_btn, down_btn, down_btn2])
-    send_btn.click(cellpose_segment, [input_image, resize], [img_outlines, img_overlay, flows, down_btn, down_btn2])
+    send_btn.click(cellpose_segment, [input_image, resize], [img_outlines, flows, down_btn, down_btn2])
 
     #down_btn.click(download_function, None, [down_btn, down_btn2])