Update app.py

app.py

@@ -8,6 +8,7 @@ import matplotlib.pyplot as plt
 import os, io, base64
 from PIL import Image 
 from cellpose.io import imread, imsave
+from cellpose.transforms import normalize_img
 
 # @title Data retrieval
 def download_weights():
@@ -223,6 +224,8 @@ def download_function():
 
 def upload_file(filepath): 
     img = imread(filepath)    
+    img = normalize_img(img)
+    img = np.clip(img, 0, 1)
     
     return img
     
@@ -238,8 +241,9 @@ with gr.Blocks(title = "Hello",
             input_image = gr.Image(label = "Input image", type = "numpy")
 
             with gr.Row():
-                up_btn = gr.UploadButton("Upload (png, jpg, tif etc)", visible=True, file_count = "single")
-                resize = gr.Number(label = 'max resize', value = 1000)                
+                with gr.Column():
+                    up_btn = gr.UploadButton("Upload (png, jpg, tif etc)", visible=True, file_count = "single")
+                    resize = gr.Number(label = 'max resize', value = 1000)                
 
                 with gr.Column():
                     send_btn = gr.Button("Run Cellpose-SAM")
@@ -262,7 +266,7 @@ with gr.Blocks(title = "Hello",
     gr.Examples(sample_list, inputs=input_image, examples_per_page=25)
     
     up_btn.upload(upload_file, up_btn, input_image)
-    send_btn.click(fn=cellpose_segment, inputs=[up_btn, resize], outputs=[img_outlines, img_overlay, flows, down_btn, down_btn2])
+    send_btn.click(cellpose_segment, [up_btn, resize], [img_outlines, img_overlay, flows, down_btn, down_btn2])
 
     #down_btn.click(download_function, None, [down_btn, down_btn2])