update app.py

app.py

@@ -774,10 +774,13 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
     use_sam_predictor = True
     if task_type == 'segment' or ((task_type in ['inpainting', 'outpainting'] or task_type == 'remove') and mask_source_radio == mask_source_segment):
         image = np.array(input_img)
+        logger.info(f'run_anything_task_[{file_temp}]_{task_type}_2_1_')
         if task_type == 'remove' and remove_use_segment == False:
+            logger.info(f'run_anything_task_[{file_temp}]_{task_type}_2_2_')
             use_sam_predictor = False
 
         if sam_predictor and use_sam_predictor:
+            logger.info(f'run_anything_task_[{file_temp}]_{task_type}_2_3_')
             sam_predictor.set_image(image)
 
         for i in range(boxes_filt.size(0)):
@@ -786,6 +789,7 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
             boxes_filt[i][2:] += boxes_filt[i][:2]
 
         if sam_predictor and use_sam_predictor:
+            logger.info(f'run_anything_task_[{file_temp}]_{task_type}_2_4_')
             boxes_filt = boxes_filt.to(sam_device)
             transformed_boxes = sam_predictor.transform.apply_boxes_torch(boxes_filt, image.shape[:2])
 
@@ -798,6 +802,7 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
             # masks: [9, 1, 512, 512]
             assert sam_checkpoint, 'sam_checkpoint is not found!'
         else:
+            logger.info(f'run_anything_task_[{file_temp}]_{task_type}_2_5_')
             masks = torch.zeros(len(boxes_filt), 1, H, W)   
             mask_count = 0         
             for box in boxes_filt:
@@ -806,6 +811,7 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
             masks = torch.where(masks > 0, True, False)      
             run_mode = "rectangle"
 
+        logger.info(f'run_anything_task_[{file_temp}]_{task_type}_2_6_')
         # draw output image
         plt.figure(figsize=(10, 10))
         plt.imshow(image)