Update app_3.py

app_3.py

@@ -1111,9 +1111,9 @@ def process_image(input_image, input_text):
             # mask_mover.set_extracted_fg(cropped_rgba)
 
             # alpha = img[..., 3] > 0
-            H, W = alpha[first_mask].shape
+            H, W = alpha.shape
             # get the bounding box of alpha
-            y, x = np.where(alpha[first_mask])
+            y, x = np.where(alpha > 0)
             y0, y1 = max(y.min() - 1, 0), min(y.max() + 1, H)
             x0, x1 = max(x.min() - 1, 0), min(x.max() + 1, W)
             
@@ -1274,9 +1274,9 @@ def process_image(input_image, input_text):
             # mask_mover.set_extracted_fg(cropped_rgba)
 
             # alpha = img[..., 3] > 0
-            H, W = alpha[first_mask].shape
+            H, W = alpha.shape
             # get the bounding box of alpha
-            y, x = np.where(alpha[first_mask])
+            y, x = np.where(alpha > 0)
             y0, y1 = max(y.min() - 1, 0), min(y.max() + 1, H)
             x0, x1 = max(x.min() - 1, 0), min(x.max() + 1, W)