Refactor yolov10_inference and yolov10_video_inference to take a single argument containing all inputs.

app.py

@@ -31,7 +31,8 @@ category_dict = {
 }
 
 @spaces.GPU(duration=200)
-def yolov10_inference(image, model_id, image_size, conf_threshold, iou_threshold):
+def yolov10_inference(inputs):
+    image, model_id, image_size, conf_threshold, iou_threshold = inputs[1], inputs[2], inputs[3], inputs[4], inputs[5]
     model_path = download_models(model_id)
     model = YOLOv10(model_path)
     results = model(source=image, imgsz=image_size, iou=iou_threshold, conf=conf_threshold, verbose=False)[0]
@@ -45,7 +46,8 @@ def yolov10_inference(image, model_id, image_size, conf_threshold, iou_threshold
 
     return annotated_image
 
-def yolov10_video_inference(video, model_id, image_size, conf_threshold, iou_threshold):
+def yolov10_video_inference(inputs):
+    video, model_id, image_size, conf_threshold, iou_threshold = inputs[2], inputs[3], inputs[4], inputs[5], inputs[6]
     model_path = download_models(model_id)
     model = YOLOv10(model_path)
 
@@ -133,8 +135,14 @@ def app():
                 output_image = gr.Image(type="numpy", label="Annotated Image", visible=True)
                 output_video = gr.Video(label="Annotated Video", visible=False)
 
+        def process_inputs(inputs):
+            if inputs[0] == "Image":
+                return yolov10_inference(inputs)
+            else:
+                return yolov10_video_inference(inputs)
+
         yolov10_infer.click(
-            fn=lambda inputs: yolov10_inference(*inputs) if inputs[0] == "Image" else yolov10_video_inference(*inputs[1:]),
+            fn=process_inputs,
             inputs=[
                 image_or_video,
                 image,
@@ -186,7 +194,7 @@ def app():
                     0.45,
                 ],
             ],
-            fn=lambda inputs: yolov10_inference(*inputs) if inputs[0] == "Image" else yolov10_video_inference(*inputs[1:]),
+            fn=process_inputs,
             inputs=[
                 image_or_video,
                 image,