zzzzz

app.py

@@ -5,7 +5,7 @@ import onnx
 import numpy as np
 import gradio as gr
 from huggingface_hub import hf_hub_download
-
+from gradio_webrtc import WebRTCStreamer
 # --- YOLOv5n ONNX 추론 클래스 (간단히 구현) ---
 import onnxruntime
 
@@ -93,16 +93,18 @@ def detection(image, weight_file, conf_threshold=0.3):
 
 # --- Gradio UI ---
 with gr.Blocks() as demo:
-    gr.Markdown("# YOLOv5n ONNX Detection with Weight Upload")
+    gr.Markdown("# YOLOv5n ONNX Detection with Weight Upload (WebRTC)")
 
     weight_file = gr.File(label="Upload YOLOv5n weights (.pt or .onnx)", file_types=['.pt', '.onnx'])
     conf_threshold = gr.Slider(0.0, 1.0, value=0.3, step=0.05, label="Confidence Threshold")
-    input_image = gr.Image(source="webcam", streaming=True)
-    output_image = gr.Image()
 
-    input_image.stream(
+    rtc_stream = WebRTCStreamer(label="WebRTC Webcam")
+
+    output_image = gr.Image(label="Detection Output")
+
+    rtc_stream.stream(
         fn=detection,
-        inputs=[input_image, weight_file, conf_threshold],
+        inputs=[rtc_stream, weight_file, conf_threshold],
         outputs=output_image,
         every=0.1,
     )