Update app.py

app.py

@@ -198,7 +198,7 @@ def get_image(img_input, mask=False, scale=True):
     if mask:
         if len(img.shape) == 3:
             img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
-        return np.where(img > 127, 1, 0)  # Threshold at 127 for the mask
+        return np.where(img > 127, 1, 0).astype(np.uint8)  # Threshold at 127 for the mask
     
     if scale and img.dtype != np.float64:
         return img.astype('float64') / 255.0
@@ -208,6 +208,9 @@ def get_image(img_input, mask=False, scale=True):
 def blend_images(bg_img, obj_img, mask_img, method):
     bg_img = get_image(bg_img)
     obj_img = get_image(obj_img)
+    
+    if isinstance(mask_img, str):
+        mask_img = cv2.imread(mask_img, cv2.IMREAD_GRAYSCALE)
     mask_img = get_image(mask_img, mask=True)
 
     if method == "Poisson":
@@ -257,11 +260,11 @@ with gr.Blocks(theme='bethecloud/storj_theme') as iface:
     )
     
     gr.Examples(
-        examples=[
-            ["img1.jpg", "img2.jpg", "mask1.jpg", "Poisson"],
-            ["img3.jpg", "img4.jpg", "mask2.jpg", "Mixed Gradient"],
-            ["img6.jpg", "img9.jpg", "mask3.jpg", "Laplacian"]
-        ],
+    examples=[
+        ["img1.jpg", "img2.jpg", "mask1.jpg", "Poisson"],
+        ["img3.jpg", "img4.jpg", "mask2.jpg", "Mixed Gradient"],
+        ["img6.jpg", "img9.jpg", "mask3.jpg", "Laplacian"]
+    ],
         inputs=[bg_img, obj_img, mask_img, method],
         outputs=output_image,
         fn=blend_images,