Update app.py

app.py

@@ -6,6 +6,7 @@ from cellpose import models
 from matplotlib.colors import hsv_to_rgb
 import matplotlib.pyplot as plt
 import os, io, base64
+from PIL import Image 
 
 try:
     model = models.CellposeModel(gpu=True, pretrained_model="cyto3")
@@ -113,6 +114,8 @@ def cellpose_segment(img_input):
     
     outpix = plot_outlines(img_input, masks)
     overlay = plot_overlay(img_input, masks)
+    masks =  Image.fromarray(masks)
+    
     return outpix, overlay, flows, masks
 
 # Gradio Interface