Update app.py

app.py

@@ -31,10 +31,10 @@ def yolo_inference(image, model_id, conf_threshold, iou_threshold):
     return annotated_image
 
 def app():
-    with gr.Blocks():
+    with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column():
-                image = gr.Image(type="pil", label="Image", tool="editor")
+                image = gr.Image(type="pil", label="Image")
                 
                 model_id = gr.Dropdown(
                     label="Model",
@@ -60,12 +60,12 @@ def app():
                     step=0.1,
                     value=0.45,
                 )
-                yolov10_infer = gr.Button(value="Detect Objects")
+                yolo_infer = gr.Button(value="Detect Objects")
 
             with gr.Column():
-                output_image = gr.Image(type="pil", label="Annotated Image", interactive=False)
+                output_image = gr.Image(type="pil", label="Annotated Image")
 
-        yolov10_infer.click(
+        yolo_infer.click(
             fn=yolo_inference,
             inputs=[
                 image,
@@ -79,7 +79,7 @@ def app():
         gr.Examples(
             examples=[
                 [
-                    "b.jpg",
+                    "a.jpg",
                     "yolov10x.pt",
                     0.25,
                     0.45,
@@ -102,10 +102,6 @@ def app():
             cache_examples="lazy",
         )
 
-gradio_app = gr.Blocks()
-with gradio_app:
-    with gr.Row():
-        with gr.Column():
-            app()
+    demo.launch(debug=True)
 
-gradio_app.launch(debug=True)
+app()