push

app.py

@@ -9,7 +9,11 @@ model = YOLOv10.from_pretrained(f'jameslahm/yolov10s')
 @spaces.GPU
 def yolov10_inference(image, conf_threshold):
     width, _ = image.size
+    import time
+    start = time.time()
     results = model.predict(source=image, imgsz=width, conf=conf_threshold)
+    end = time.time()
+    print("time", end - start)
     annotated_image = results[0].plot()
     return annotated_image[:, :, ::-1]
 
@@ -18,7 +22,7 @@ def app():
     with gr.Blocks():
         with gr.Row():
             with gr.Column():
-                image = gr.Image(type="pil", label="Image", visible=True)
+                image = gr.Image(type="pil", label="Image", visible=True, sources="webcam", height=500, width=500)
                 conf_threshold = gr.Slider(
                     label="Confidence Threshold",
                     minimum=0.0,
@@ -26,10 +30,6 @@ def app():
                     step=0.05,
                     value=0.25,
                 )
-
-            with gr.Column():
-                output_image = gr.Image(type="numpy", label="Annotated Image", visible=True)
-
         image.stream(
             fn=yolov10_inference,
             inputs=[image, conf_threshold],