playing around with cv2

app.py

@@ -194,13 +194,12 @@ def detections_to_json(detections):
 def process_image(image):
     labels = ["insect"]
     original_image, detections = grounded_segmentation(image, labels, threshold=0.3, polygon_refinement=True)
-    annotated_image = plot_detections(original_image, detections)
     yellow_background_with_insects = create_yellow_background_with_insects(np.array(original_image), detections)
     detections_json = detections_to_json(detections)
     json_output_path = "insect_detections.json"
     with open(json_output_path, 'w') as json_file:
         json.dump(detections_json, json_file, indent=4)
-    return annotated_image, json.dumps(detections_json, separators=(',', ':'))
+    return yellow_background_with_insects, json.dumps(detections_json, separators=(',', ':'))
 
 gr.Interface(
     fn=process_image,