Update app.py

PIL image test

app.py

@@ -92,14 +92,15 @@ def image_resize(img, resize=400):
     return img
 
 
-@spaces.GPU(duration=10)
-def run_model_gpu(model, img):
-    masks, flows, _ = model.eval(img, channels=[0,0])
-    return masks, flows
+#@spaces.GPU(duration=10)
+#def run_model_gpu(model, img):
+#    masks, flows, _ = model.eval(img, channels=[0,0])
+ #   return masks, flows
 
 def cellpose_segment(img_input):
     img = image_resize(img_input)
-    masks, flows = run_model_gpu(model, img)
+    #masks, flows = run_model_gpu(model, img)
+    masks, flows, _ = model.eval(img, channels=[0,0])
     flows = flows[0]
     # masks = np.zeros(img.shape[:2])
     # flows = np.zeros_like(img)
@@ -111,7 +112,7 @@ def cellpose_segment(img_input):
     
     outpix = plot_outlines(img_input, masks)
     overlay = plot_overlay(img_input, masks)
-    return outpix, overlay, flows, masks 
+    return outpix, overlay, flows, Image.fromarray(masks ) 
 
 # Gradio Interface
 iface = gr.Interface(