Update handler.py

handler.py

@@ -504,14 +504,20 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
         for box, label in zip(boxes_filt, pred_phrases):
             show_box(box.cpu().numpy(), plt.gca(), label)
         plt.axis('off')
-        image_path = os.path.join(output_dir, f"grounding_seg_output_{file_temp}.jpg")
-        plt.savefig(image_path, bbox_inches="tight")
+
+        # Save the plot to a BytesIO object in memory
+        buf = io.BytesIO()
+        plt.savefig(buf, format='jpeg', bbox_inches='tight')
+        buf.seek(0)
+
+        # Convert the image in memory to a PIL Image
+        segment_image_result = Image.open(buf).convert('RGB')
+        output_images.append(segment_image_result)
+
+        # Clearing memory
+        buf.close()
         plt.clf()
-        plt.close('all')
-        segment_image_result = cv2.cvtColor(cv2.imread(image_path), cv2.COLOR_BGR2RGB)
-        os.remove(image_path)
-        output_images.append(Image.fromarray(segment_image_result)) 
-        run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)       
+        plt.close('all')     
 
     print(sam_predictor)