update app.py

app.py

@@ -739,9 +739,6 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
     size = image_pil.size
     H, W = size[1], size[0]
 
-    if remove_use_segment == False and task_type == 'remove':
-        remove_mode = 'rectangle'
-
     # run grounding dino model
     if (task_type in ['inpainting', 'outpainting'] or task_type == 'remove') and mask_source_radio == mask_source_draw:
         pass
@@ -774,10 +771,13 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
 
     logger.info(f'run_anything_task_[{file_temp}]_{task_type}_2_')
 
-    if task_type == 'segment' or ((task_type in ['inpainting', 'outpainting'] or (task_type == 'remove' and remove_use_segment )) and mask_source_radio == mask_source_segment):
+    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)
+        if task_type == 'remove' and remove_use_segment == False:
+            use_sam_predictor = False
 
-        if sam_predictor:
+        if sam_predictor and use_sam_predictor:
             sam_predictor.set_image(image)
 
         for i in range(boxes_filt.size(0)):
@@ -785,7 +785,7 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
             boxes_filt[i][:2] -= boxes_filt[i][2:] / 2
             boxes_filt[i][2:] += boxes_filt[i][:2]
 
-        if sam_predictor:
+        if sam_predictor and use_sam_predictor:
             boxes_filt = boxes_filt.to(sam_device)
             transformed_boxes = sam_predictor.transform.apply_boxes_torch(boxes_filt, image.shape[:2])