Update app.py

app.py

@@ -99,6 +99,10 @@ def image_resize(img, resize=400):
     return img
 
 
+def plot_masks(masks):
+    
+    return RGB
+    
 #@spaces.GPU(duration=10)
 #def run_model_gpu(model, img):
 #    masks, flows, _ = model.eval(img, channels=[0,0])
@@ -126,7 +130,8 @@ def cellpose_segment(img_input):
 
     overlay = Image.fromarray(overlay)
     flows = Image.fromarray(flows)
-    masks = Image.fromarray(crand[masks])
+    
+    masks = Image.fromarray(255. * crand[masks])
     
     return outpix, overlay, flows, masks
 
@@ -148,8 +153,8 @@ with gr.Blocks(title = "Hello",
             send_btn = gr.Button("Run Cellpose-SAM")
 
         with gr.Column(scale=1):            
-            img_overlay = gr.Image(label = "Output image", type = "pil")
             img_outlines = gr.Image(label = "Output image", type = "pil")            
+            img_overlay = gr.Image(label = "Output image", type = "pil")            
             flows = gr.Image(label = "Output image", type = "pil")
             masks = gr.Image(label = "Output image", type = "pil")