Update app.py

app.py

@@ -121,11 +121,12 @@ def cellpose_segment(img_input):
     outpix = plot_outlines(img_input, masks)
     overlay = plot_overlay(img_input, masks)
 
-    iperm = np.random.permutation(np.max(masks.flatten()).astype('int')+1)
+    crand = .2 + .8 * np.random.rand(np.max(masks.flatten()).astype('int')+1, 'float32')
+    crand[0] = 0
 
     overlay = Image.fromarray(overlay)
     flows = Image.fromarray(flows)
-    masks = Image.fromarray(iperm[masks])
+    masks = Image.fromarray(crand[masks])
     
     return outpix, overlay, flows, masks