gsl_result app.py

app.py

@@ -1,7 +1,7 @@
 try:
     import detectron2
 except:
-    import os 
+    import os
     os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
 
 import gradio as gr
@@ -10,12 +10,15 @@ import numpy as np
 from sam_utils import grounded_segmentation, create_yellow_background_with_insects
 from yolo_utils import yolo_processimage
 from detectron_utils import detectron_process_image
+from gsl_utils import gsl_process_image
+
 def process_image(image, include_json):
-    detectron_result=detectron_process_image(image)
+    detectron_result = detectron_process_image(image)
     yolo_result = yolo_processimage(image)
     insectsam_result = create_yellow_background_with_insects(image)
+    gsl_result = gsl_process_image(image)
 
-    return insectsam_result, yolo_result, detectron_result
+    return insectsam_result, yolo_result, detectron_result, gsl_result
 
 examples = [
     ["demo.jpg"],
@@ -29,9 +32,12 @@ examples = [
 gr.Interface(
     fn=process_image,
     inputs=[gr.Image(type="pil")],
-    outputs=[gr.Image(label='InsectSAM', type="numpy"),
-             gr.Image(label='Yolov8', type="numpy"),
-             gr.Image(label='Detectron', type="numpy")],
+    outputs=[
+        gr.Image(label='InsectSAM', type="numpy"),
+        gr.Image(label='Yolov8', type="numpy"),
+        gr.Image(label='Detectron', type="numpy"),
+        gr.Image(label='GSL', type="numpy")
+    ],
     title="Insect Model Zoo 🐞🔬",
     examples=examples
 ).launch()