Update app.py

app.py

@@ -48,13 +48,17 @@ def yolov10_inference_multi(image, image_size, conf_threshold, iou_threshold):
     yolov10s_image = yolov10_inference(image, "yolov10s.pt", image_size, conf_threshold, iou_threshold)
     yolov10m_image = yolov10_inference(image, "yolov10m.pt", image_size, conf_threshold, iou_threshold)
     yolov10b_image = yolov10_inference(image, "yolov10b.pt", image_size, conf_threshold, iou_threshold)
-    return yolov10n_image, yolov10s_image, yolov10m_image, yolov10b_image
+    yolov10l_image = yolov10_inference(image, "yolov10l.pt", image_size, conf_threshold, iou_threshold)
+    yolov10x_image = yolov10_inference(image, "yolov10x.pt", image_size, conf_threshold, iou_threshold)
+    return yolov10n_image, yolov10s_image, yolov10m_image, yolov10b_image, yolov10l_image, yolov10x_image
 
 def app():
     with gr.Blocks():
         with gr.Row():
             with gr.Column():
                 image = gr.Image(type="pil", label="Image")
+                output_image_l = gr.Image(type="pil", label="yolov10l")
+                output_image_x = gr.Image(type="pil", label="yolov10x")
                 image_size = gr.Slider(
                     label="Image Size",
                     minimum=320,
@@ -92,7 +96,7 @@ def app():
                 conf_threshold,
                 iou_threshold,
             ],
-            outputs=[output_image_n, output_image_s, output_image_m, output_image_b],
+            outputs=[output_image_n, output_image_s, output_image_m, output_image_b, , output_image_l, output_image_x],
         )
 
         gr.Examples(
@@ -123,7 +127,7 @@ def app():
                 conf_threshold,
                 iou_threshold,
             ],
-            outputs=[output_image_n, output_image_s, output_image_m, output_image_b],
+            outputs=[output_image_n, output_image_s, output_image_m, output_image_b, , output_image_l, output_image_x],
             cache_examples=True,
         )
 
@@ -132,7 +136,7 @@ with gradio_app:
     gr.HTML(
         """
     <h1 style='text-align: center'>
-    YOLOv10: Real-Time Object Detection
+    YOLOv10 - Comparison of Models
     </h1>
     """)
     with gr.Row():