From 8dfb847bc9c2867c46515a72ed7b32659d69f6db Mon Sep 17 00:00:00 2001
From: Hich Tala <98521878+HichTala@users.noreply.github.com>
Date: Fri, 9 Feb 2024 09:37:28 +0100
Subject: [PATCH] Update config.json

---
 config.json | 1 +
 draw.py     | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/config.json b/config.json
index dcff13f..5e9373e 100644
--- a/config.json
+++ b/config.json
@@ -1,6 +1,7 @@
 {
   "trained_models": "/models",
   "data_path": "/data",
+  "webcam": 0,
   "card_types": ["spell", "monster", "normal_monster", "trap", "fusion", "ritual"],
   "monster": {
     "colors": 0,
diff --git a/draw.py b/draw.py
index 9e9f9d1..155927f 100644
--- a/draw.py
+++ b/draw.py
@@ -28,6 +28,9 @@ def __init__(self, config, deck_list, source, debug=False):
             device=device
         )
 
+        if source == '0':
+            source = str(self.configs['webcam'])
+
         self.results = model_regression(
             source=source,
             show_labels=False,