Update app.py

app.py

@@ -132,7 +132,8 @@ def run_model_gpu(img):
     return masks, flows
 
 #@spaces.GPU(duration=10)
-def cellpose_segment(img_input):
+def cellpose_segment(img_pil):
+    img_input = np.array(img_pil)
     img = image_resize(img_input)
     masks, flows = run_model_gpu(img)
     #masks, flows, _ = model.eval(img, channels=[0,0])
@@ -166,8 +167,9 @@ def cellpose_segment(img_input):
 
     pil_masks = Image.fromarray(masks.astype('int32'))
     pil_masks.save("masks.tiff")
+    
 
-    outpix.save("outlines.png")
+    outpix.save(img_pil.filename) #"outlines.png")
 
     b1 = gr.DownloadButton(visible=True, value = "masks.tiff")
     b2 = gr.DownloadButton(visible=True, value = "outlines.png")
@@ -197,7 +199,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 time/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 = "pil")
             send_btn = gr.Button("Run Cellpose-SAM")
             with gr.Row():
                 down_btn = gr.DownloadButton("Download masks (TIFF)", visible=False)