Use the same threshold for both models

app.py

@@ -67,10 +67,8 @@ def run_yolov8(image_path: str, threshold: float) -> np.ndarray:
     return res
 
 
-def run(image_path: str, threshold_deta: float,
-        threshold_yolo: float) -> tuple[np.ndarray, np.ndarray]:
-    return run_deta(image_path,
-                    threshold_deta), run_yolov8(image_path, threshold_yolo)
+def run(image_path: str, threshold: float) -> tuple[np.ndarray, np.ndarray]:
+    return run_deta(image_path, threshold), run_yolov8(image_path, threshold)
 
 
 with gr.Blocks(css='style.css') as demo:
@@ -78,16 +76,11 @@ with gr.Blocks(css='style.css') as demo:
     with gr.Row():
         with gr.Column():
             image = gr.Image(label='Input image', type='filepath')
-            threshold_deta = gr.Slider(label='Score threshold for DETA',
-                                       minimum=0,
-                                       maximum=1,
-                                       value=0.1,
-                                       step=0.01)
-            threshold_yolo = gr.Slider(label='Score threshold for YOLOv8',
-                                       minimum=0,
-                                       maximum=1,
-                                       value=0.5,
-                                       step=0.01)
+            threshold = gr.Slider(label='Score threshold',
+                                  minimum=0,
+                                  maximum=1,
+                                  value=0.5,
+                                  step=0.01)
             run_button = gr.Button('Run')
         with gr.Column():
             result_deta = gr.Image(label='Result (DETA)', type='numpy')
@@ -95,11 +88,10 @@ with gr.Blocks(css='style.css') as demo:
 
     with gr.Row():
         paths = sorted(pathlib.Path('images').glob('*.jpg'))
-        gr.Examples(examples=[[path.as_posix(), 0.1, 0.5] for path in paths],
+        gr.Examples(examples=[[path.as_posix(), 0.5] for path in paths],
                     inputs=[
                         image,
-                        threshold_deta,
-                        threshold_yolo,
+                        threshold,
                     ],
                     outputs=[
                         result_deta,
@@ -111,8 +103,7 @@ with gr.Blocks(css='style.css') as demo:
     run_button.click(fn=run,
                      inputs=[
                          image,
-                         threshold_deta,
-                         threshold_yolo,
+                         threshold,
                      ],
                      outputs=[
                          result_deta,