toecm commited on
Commit
76c71e9
·
verified ·
1 Parent(s): 7c6553f

Update auto_train_watcher.py

Browse files
Files changed (1) hide show
  1. auto_train_watcher.py +2 -2
auto_train_watcher.py CHANGED
@@ -5,7 +5,7 @@ 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
11
 
@@ -49,7 +49,7 @@ def trigger_autotrain(data_count):
49
  client = AutoTrainClient(hf_token=HF_TOKEN)
50
 
51
  # Create project
52
- project_name = f"pure-convo-finetune-{int(time.time())}"
53
  client.create_project(project_name, task="speech-recognition")
54
 
55
  # Add data & Start
 
5
 
6
  # CONFIGURATION
7
  DATASET_ID = "toecm/IEDID" # Where the app saves data
8
+ MODEL_ID = "toecm/PureVersation_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
11
 
 
49
  client = AutoTrainClient(hf_token=HF_TOKEN)
50
 
51
  # Create project
52
+ project_name = f"pure-versation-finetune-{int(time.time())}"
53
  client.create_project(project_name, task="speech-recognition")
54
 
55
  # Add data & Start