update app.py

app.py

@@ -773,13 +773,11 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
         run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
 
     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 mask_source_radio == mask_source_segment):
+
+    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):
         image = np.array(input_img)
-        use_sam_predictor = True
-        if remove_use_segment and task_type == 'remove':
-            use_sam_predictor = False
 
-        if sam_predictor and use_sam_predictor:
+        if sam_predictor:
             sam_predictor.set_image(image)
 
         for i in range(boxes_filt.size(0)):
@@ -787,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 and use_sam_predictor:
+        if sam_predictor:
             boxes_filt = boxes_filt.to(sam_device)
             transformed_boxes = sam_predictor.transform.apply_boxes_torch(boxes_filt, image.shape[:2])