Update app.py

app.py

@@ -164,8 +164,8 @@ def run_model_gpu1000(img):
     return masks, flows
 
 #@spaces.GPU(duration=10)
-def cellpose_segment(img_pil, resize = 1000):
-    img_input = imread(img_pil)
+def cellpose_segment(filepath, resize = 1000):
+    img_input = imread(filepath[0])
     print(img_input.shape)
     #img_input = np.array(img_pil)
     img = image_resize(img_input, resize = resize)
@@ -213,8 +213,8 @@ def cellpose_segment(img_pil, resize = 1000):
     #pil_masks = Image.fromarray(masks.astype('int32'))
     #pil_masks.save(fname_mask)
 
-    fname_out  = os.path.splitext(img_pil)[0]+"_outlines.png"
-    fname_masks = os.path.splitext(img_pil)[0]+"_masks.tif"
+    fname_out  = os.path.splitext(filepath[0])[0]+"_outlines.png"
+    fname_masks = os.path.splitext(filepath[0])[0]+"_masks.tif"
 
     imsave(fname_masks, masks)
     
@@ -250,7 +250,7 @@ def upload_file(filepath):
 
     #b1 = gr.DownloadButton("Download masks as TIFF", visible=False)
     #b2 = gr.DownloadButton("Download outline image as PNG", visible=False)
-    return filepath #, b1, b2
+    return [filepath] #, b1, b2
 
  
 def update_image(filepath): 
@@ -260,7 +260,7 @@ def update_image(filepath):
 
     #b1 = gr.DownloadButton("Download masks as TIFF", visible=False)
     #b2 = gr.DownloadButton("Download outline image as PNG", visible=False)
-    return filepath#, b1, b2  
+    return filepath[0]#, b1, b2  
     
 with gr.Blocks(title = "Hello", 
                css=".gradio-container {background:purple;}") as demo: