toecm commited on
Commit
ef938eb
·
verified ·
1 Parent(s): 49f1739

Update auto_train_watcher.py

Browse files
Files changed (1) hide show
  1. auto_train_watcher.py +1 -1
auto_train_watcher.py CHANGED
@@ -4,7 +4,7 @@ from huggingface_hub import HfApi, snapshot_download
4
  from datasets import load_dataset
5
 
6
  # CONFIGURATION
7
- DATASET_ID = "toecm/PureChain_Dataset" # Where the app saves data
8
  MODEL_ID = "toecm/PureConvo_Model" # The model to train
9
  THRESHOLD = 1000 # How many new clips before training?
10
  HF_TOKEN = os.environ.get("HF_TOKEN") # Your Write Access Token
 
4
  from datasets import load_dataset
5
 
6
  # CONFIGURATION
7
+ DATASET_ID = "toecm/IEDID" # Where the app saves data
8
  MODEL_ID = "toecm/PureConvo_Model" # The model to train
9
  THRESHOLD = 1000 # How many new clips before training?
10
  HF_TOKEN = os.environ.get("HF_TOKEN") # Your Write Access Token