Update app.py

app.py

@@ -26,7 +26,7 @@ def yolov10_inference(image, model_id, image_size, conf_threshold, iou_threshold
     return Image.fromarray(annotated_image[..., ::-1])
 
 def app():
-    with gr.Blocks():
+    with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column():
                 image = gr.Image(type="pil", label="Image")
@@ -67,23 +67,18 @@ 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=[
-                ["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",
+            examples=["Rocket.png"],
+            inputs=[image],
         )
+    return demo
 
 if __name__ == "__main__":
-    app().launch()
+    app().launch()