Update app.py

app.py

@@ -13,8 +13,9 @@ import glob
 from huggingface_hub import hf_hub_download
 
 img = np.zeros((96, 128))
-fp0 = "0.png"
-imsave(fp0, img)
+fp0 = Image.fromarray(img)
+#fp0 = "0.png"
+#imsave(fp0, img)
 
 # data  retrieval
 def download_weights():    
@@ -224,13 +225,8 @@ def cellpose_segment(filepath, resize = 1000,max_iter = 250):
     flows = flows.resize((Lx, Ly), resample  = Image.BICUBIC)
 
     fname_out  = os.path.splitext(filepath[-1])[0]+"_outlines.png"
-    try:
-        os.remove(fname_out)
-    except:
-        print('not there')
     outpix.save(fname_out) #"outlines.png")
-    print(fname_out)
-
+    
     fname_flows  = os.path.splitext(filepath[-1])[0]+"_flows.png"
     flows.save(fname_flows) #"outlines.png")
 
@@ -240,7 +236,7 @@ def cellpose_segment(filepath, resize = 1000,max_iter = 250):
         b1 = gr.DownloadButton(visible=True, value = fname_masks)
     b2 = gr.DownloadButton(visible=True, value = fname_out) #"outlines.png")
     
-    return fname_out, fname_flows, b1, b2
+    return outpix, flows, b1, b2
 
 # Gradio Interface
 #iface = gr.Interface(
@@ -330,9 +326,9 @@ with gr.Blocks(title = "Hello",
                     down_btn2 = gr.DownloadButton("Download outlines (PNG)", visible=False)  
                     
         with gr.Column(scale=2):     
-            outlines = gr.Image(label = "Outlines", type = "filepath", format = 'png', value = fp0) #, width = "50vw", height = "20vw")
+            outlines = gr.Image(label = "Outlines", type = "pil", format = 'png', value = fp0) #, width = "50vw", height = "20vw")
             #img_overlay = gr.Image(label = "Overlay", type = "pil", format = 'png') #, width = "50vw", height = "20vw")
-            flows = gr.Image(label = "Cellpose flows", type = "filepath", format = 'png', value = fp0) #, width = "50vw", height = "20vw")
+            flows = gr.Image(label = "Cellpose flows", type = "pil", format = 'png', value = fp0) #, width = "50vw", height = "20vw")