Spaces:

Collov-Labs
/

d-edit

Running on Zero

niulx commited on Sep 24

Commit

6cf97e1

•

1 Parent(s): 842da73

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -87,7 +87,7 @@ def transparent_paste_with_mask(backimg, foreimg, mask_np,transparency = 128):
     mask_np = mask_np[:,:,np.newaxis]
     new_img_np = bimg_np*mask_np + (1-mask_np)* backimg_solid_np
-    return Image.fromarray(new_img_np)
 def show_segmentation(image, segmentation, flag):
     if flag is False:

     mask_np = mask_np[:,:,np.newaxis]
     new_img_np = bimg_np*mask_np + (1-mask_np)* backimg_solid_np
+    return Image.fromarray(np.uint8(new_img_np))
 def show_segmentation(image, segmentation, flag):
     if flag is False: