Update app.py

app.py

@@ -53,6 +53,9 @@ def detect_webcam(camera_input, conf_threshold):
     if current_model is None:
         raise gr.Error("请先上传模型文件")
     
+    if camera_input is None:
+        return None
+    
     results = current_model(camera_input, conf=conf_threshold)
     plotted = results[0].plot()
     return plotted[:, :, ::-1]  # BGR转RGB
@@ -82,14 +85,15 @@ with gr.Blocks() as demo:
             video_button = gr.Button("执行检测")
         
         with gr.TabItem("实时摄像头"):
-            webcam_input = gr.Image(label="摄像头画面", source="webcam", streaming=True)
-            webcam_output = gr.Image(label="检测结果", streaming=True)
+            webcam_input = gr.Webcam(label="摄像头画面")
+            webcam_output = gr.Image(label="检测结果")
             webcam_conf = gr.Slider(0, 1, value=0.5, label="置信度阈值")
+            webcam_button = gr.Button("开始检测")
     
     # 绑定事件处理
     img_button.click(fn=detect_image, inputs=[img_input, img_conf], outputs=img_output)
     video_button.click(fn=detect_video, inputs=[video_input, video_conf], outputs=video_output)
-    webcam_input.stream(fn=detect_webcam, inputs=[webcam_input, webcam_conf], outputs=webcam_output)
+    webcam_button.click(fn=detect_webcam, inputs=[webcam_input, webcam_conf], outputs=webcam_output)
 
 if __name__ == "__main__":
     demo.launch()