Update app_3.py

app_3.py

@@ -1404,9 +1404,9 @@ def process_image(input_image, input_text):
             alpha = np.zeros((H, W, 1), dtype=np.uint8)
             alpha[~first_mask] = 128 # Set semi-transparency for background
             alpha[first_mask] = 255 # Make the foreground opaque
-
+            alpha = alpha.squeeze(-1) # Remove singleton dimension to become 2D
             rgba = np.dstack((img, alpha)).astype(np.uint8)
-            
+
             # get the bounding box of alpha
             y, x = np.where(alpha > 0)
             y0, y1 = max(y.min() - 1, 0), min(y.max() + 1, H)
@@ -1503,9 +1503,8 @@ def process_image(input_image, input_text):
             alpha = np.zeros((H, W, 1), dtype=np.uint8)
             alpha[~first_mask] = 128 # Set semi-transparency for background
             alpha[first_mask] = 255 # Make the foreground opaque
-
+            alpha = alpha.squeeze(-1) # Remove singleton dimension to become 2D
             rgba = np.dstack((img, alpha)).astype(np.uint8)
-            
             # get the bounding box of alpha
             y, x = np.where(alpha > 0)
             y0, y1 = max(y.min() - 1, 0), min(y.max() + 1, H)