diff --git a/examples/tree_demo/tree_demo.py b/examples/tree_demo/tree_demo.py
index ba2659f..6ab8214 100644
--- a/examples/tree_demo/tree_demo.py
+++ b/examples/tree_demo/tree_demo.py
@@ -38,9 +38,10 @@
     parser.add_argument("--image_quality", type=int, default=50)
     parser.add_argument("--port", type=int, default=7860)
     parser.add_argument("--host", type=str, default="0.0.0.0")
+    parser.add_argument("--camera", type=int, default=0)
     args = parser.parse_args()
 
-    CAMERA_DEVICE = 0
+    CAMERA_DEVICE = args.camera
     IMAGE_QUALITY = args.image_quality
 
     predictor = TreePredictor(