Update app.py

app.py

@@ -8,7 +8,6 @@ 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():
@@ -224,7 +223,7 @@ def download_function():
 
 def upload_file(filepath): 
     img = imread(filepath)    
-    img = normalize_img(img)
+    img = normalize99(img)
     img = np.clip(img, 0, 1)
     
     return img
@@ -241,11 +240,11 @@ with gr.Blocks(title = "Hello",
             input_image = gr.Image(label = "Input image", type = "numpy")
 
             with gr.Row():
-                with gr.Column():
+                with gr.Column(scale=1):
                     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():
+                with gr.Column(scale=1):
                     send_btn = gr.Button("Run Cellpose-SAM")
                     down_btn = gr.DownloadButton("Download masks (TIF)", visible=False)            
                     down_btn2 = gr.DownloadButton("Download outlines (PNG)", visible=False)