Update app.py

app.py

@@ -7,7 +7,7 @@ from matplotlib.colors import hsv_to_rgb
 import matplotlib.pyplot as plt
 import os, io, base64
 from PIL import Image 
-
+from cellpose.io import imread
 
 # @title Data retrieval
 def download_weights():
@@ -133,7 +133,8 @@ def run_model_gpu(img):
 
 #@spaces.GPU(duration=10)
 def cellpose_segment(img_pil):
-    img_input = np.array(img_pil)
+    img_input = imread(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])
@@ -199,7 +200,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.ImageEditor(label = "Input image")#, type = "pil")
+            input_image = gr.Image(label = "Input image", type = "str")
             send_btn = gr.Button("Run Cellpose-SAM")
             with gr.Row():
                 down_btn = gr.DownloadButton("Download masks (TIFF)", visible=False)