Update app.py

app.py

@@ -233,9 +233,12 @@ def upload_file(filepath):
     img = normalize99(img)
     img = np.clip(img, 0, 1)
 
+    filegui  = os.path.splitext(filepath)[0]+"_gui.png"
+    imsave(filegui, img)
+
     b1 = gr.DownloadButton("Download masks as TIFF", visible=False)
     b2 = gr.DownloadButton("Download outline image as PNG", visible=False)
-    return img, filepath, b1, b2
+    return filegui, filepath, b1, b2
 
  
 def update_image(filepath): 
@@ -257,7 +260,7 @@ with gr.Blocks(title = "Hello",
             gr.HTML("""<h4 style="color:white;">You may need to refresh/login for 5 minutes of free GPU compute/day. </h4>""")
             gr.HTML("""<h4 style="color:white;">"pip install cellpose" for full functionality. </h4>""")
 
-            input_image = gr.Image(label = "Input image", type = "numpy")
+            input_image = gr.Image(label = "Input image", type = "filepath")
 
             with gr.Row():
                 with gr.Column(scale=1):