Update app.py

app.py

@@ -67,60 +67,23 @@ def app():
 
             with gr.Column():
                 output_image = gr.Image(type="pil", label="Annotated Image")
+        
+        yolov10_infer.click(
+            fn=yolov10_inference,
+            inputs=[image, model_id, image_size, conf_threshold, iou_threshold],
+            outputs=[output_image],
+        )
 
-    gr.Examples(
-        examples=[
-            ["dog.jpeg", "yolov10m", 640, 0.25, 0.45],
-            ["huggingface.jpg", "yolov10m", 640, 0.25, 0.45],
-            ["zidane.jpg", "yolov10m", 640, 0.25, 0.45],
-        ],
-        fn=yolov10_inference,
-        inputs=[image, model_id, image_size, conf_threshold, iou_threshold],
-        outputs=[output_image],
-        cache_examples='lazy',
-    )
-
-    yolov10_infer.click(
-        fn=yolov10_inference,
-        inputs=[image, model_id, image_size, conf_threshold, iou_threshold],
-        outputs=[output_image],
-    )
-
-gradio_app = gr.Blocks()
-
-with gradio_app:
-    with gr.Row():
-        with gr.Column():
-            gr.HTML(
-                """
-            <h1 style='text-align: center'>
-            YOLOv10: Real-Time End-to-End Object Detection
-            </h1>
-            """
-            )
-            gr.HTML(
-                """
-                <h3 style='text-align: center'>
-                <a href='https://arxiv.org/abs/2405.14458' target='_blank'>arXiv</a> | <a href='https://github.com/THU-MIG/yolov10' target='_blank'>github</a>
-                </h3>
-                """
-            )
-            gr.HTML(
-                """
-            <h3 style='text-align: center'>
-            Follow me for more!
-            </h3>
-            """
-            )
-            gr.HTML(
-                """
-                <div style='text-align: center;'>
-                Twitter | Github | Linkedin | HuggingFace
-                </div>
-                """
-            )
-
-    app()
+        gr.Examples(
+            examples=[
+                ["zidane.jpg", "yolov10s", 640, 0.25, 0.45],
+                ["dog.jpeg", "yolov10s", 640, 0.25, 0.45],
+            ],
+            inputs=[image, model_id, image_size, conf_threshold, iou_threshold],
+            outputs=[output_image],
+            fn=yolov10_inference,
+            cache_examples="lazy",
+        )
 
 if __name__ == "__main__":
-    gradio_app.launch()
+    app().launch()