update app.py

app.py

@@ -1,4 +1,7 @@
 
+import warnings
+warnings.filterwarnings('ignore')
+
 import subprocess, io, os, sys, time
 from loguru import logger
 
@@ -335,7 +338,7 @@ def run_grounded_sam(input_image, text_prompt, task_type, inpaint_prompt, box_th
             return [], gr.Gallery.update(label='Please upload a image!😂😂😂😂')
 
     file_temp = int(time.time())
-    logger.info(f'run_grounded_sam_[{file_temp}]_{task_type}_[{text_prompt}]_1_')
+    logger.info(f'run_grounded_sam_[{file_temp}]_{task_type}/{inpaint_mode}/[{mask_source_radio}]/{remove_mode}/{remove_mask_extend}_[{text_prompt}]/[{inpaint_prompt}]___1_')
 
     # load image
     input_mask_pil = input_image['mask']
@@ -431,6 +434,7 @@ def run_grounded_sam(input_image, text_prompt, task_type, inpaint_prompt, box_th
             mask_pil = input_mask_pil
             mask = input_mask          
         else:
+            masks_ori = copy.deepcopy(masks)
             if inpaint_mode == 'merge':
                 masks = torch.sum(masks, dim=0).unsqueeze(0)
                 masks = torch.where(masks > 0, True, False)
@@ -452,7 +456,7 @@ def run_grounded_sam(input_image, text_prompt, task_type, inpaint_prompt, box_th
             # remove from mask
             if mask_source_radio == mask_source_segment:
                 mask_imgs = []
-                masks_shape = masks.shape        
+                masks_shape = masks_ori.shape        
                 boxes_filt_ori_array = boxes_filt_ori.numpy()
                 if inpaint_mode == 'merge':
                     extend_shape_0 = masks_shape[0]
@@ -462,7 +466,7 @@ def run_grounded_sam(input_image, text_prompt, task_type, inpaint_prompt, box_th
                     extend_shape_1 = 1
                 for i in range(extend_shape_0):
                     for j in range(extend_shape_1):                
-                        mask = masks[i][j].cpu().numpy()
+                        mask = masks_ori[i][j].cpu().numpy()
                         mask_pil = Image.fromarray(mask)
                     
                         if remove_mode == 'segment':
@@ -477,7 +481,7 @@ def run_grounded_sam(input_image, text_prompt, task_type, inpaint_prompt, box_th
                         mask_pil_exp = mask_extend(copy.deepcopy(mask_pil).convert("RGB"), 
                                         xywh_to_xyxy(torch.tensor(boxes_filt_ori_array[i]), size[0], size[1]),
                                         extend_pixels=remove_mask_extend, useRectangle=useRectangle)
-                        mask_imgs.append(mask_pil_exp)        
+                        mask_imgs.append(mask_pil_exp)
                 mask_pil = mix_masks(mask_imgs)
 
                 image_path = os.path.join(output_dir, f"image_mask_{file_temp}.jpg")