Spaces:
Running
Running
Update auto_train_watcher.py
Browse files- 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/
|
| 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-
|
| 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
|