Spaces:
Runtime error
Runtime error
Store dataset ID in AutoTrain project name
Browse files
app.py
CHANGED
|
@@ -1,5 +1,4 @@
|
|
| 1 |
import os
|
| 2 |
-
import uuid
|
| 3 |
from pathlib import Path
|
| 4 |
|
| 5 |
import pandas as pd
|
|
@@ -13,6 +12,7 @@ from evaluation import filter_evaluated_models
|
|
| 13 |
from utils import (
|
| 14 |
AUTOTRAIN_TASK_TO_HUB_TASK,
|
| 15 |
commit_evaluation_log,
|
|
|
|
| 16 |
format_col_mapping,
|
| 17 |
get_compatible_models,
|
| 18 |
get_dataset_card_url,
|
|
@@ -459,10 +459,9 @@ with st.form(key="form"):
|
|
| 459 |
)
|
| 460 |
print("INFO -- Selected models after filter:", selected_models)
|
| 461 |
if len(selected_models) > 0:
|
| 462 |
-
project_id = str(uuid.uuid4())[:8]
|
| 463 |
project_payload = {
|
| 464 |
"username": AUTOTRAIN_USERNAME,
|
| 465 |
-
"proj_name":
|
| 466 |
"task": TASK_TO_ID[selected_task],
|
| 467 |
"config": {
|
| 468 |
"language": AUTOTRAIN_TASK_TO_LANG[selected_task]
|
|
|
|
| 1 |
import os
|
|
|
|
| 2 |
from pathlib import Path
|
| 3 |
|
| 4 |
import pandas as pd
|
|
|
|
| 12 |
from utils import (
|
| 13 |
AUTOTRAIN_TASK_TO_HUB_TASK,
|
| 14 |
commit_evaluation_log,
|
| 15 |
+
create_autotrain_project_name,
|
| 16 |
format_col_mapping,
|
| 17 |
get_compatible_models,
|
| 18 |
get_dataset_card_url,
|
|
|
|
| 459 |
)
|
| 460 |
print("INFO -- Selected models after filter:", selected_models)
|
| 461 |
if len(selected_models) > 0:
|
|
|
|
| 462 |
project_payload = {
|
| 463 |
"username": AUTOTRAIN_USERNAME,
|
| 464 |
+
"proj_name": create_autotrain_project_name(selected_dataset),
|
| 465 |
"task": TASK_TO_ID[selected_task],
|
| 466 |
"config": {
|
| 467 |
"language": AUTOTRAIN_TASK_TO_LANG[selected_task]
|
utils.py
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
import inspect
|
|
|
|
| 2 |
from typing import Dict, List, Union
|
| 3 |
|
| 4 |
import jsonlines
|
|
@@ -180,3 +181,12 @@ def get_dataset_card_url(dataset_id: str) -> str:
|
|
| 180 |
return f"https://huggingface.co/datasets/{dataset_id}/edit/main/README.md"
|
| 181 |
else:
|
| 182 |
return f"https://github.com/huggingface/datasets/edit/master/datasets/{dataset_id}/README.md"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import inspect
|
| 2 |
+
import uuid
|
| 3 |
from typing import Dict, List, Union
|
| 4 |
|
| 5 |
import jsonlines
|
|
|
|
| 181 |
return f"https://huggingface.co/datasets/{dataset_id}/edit/main/README.md"
|
| 182 |
else:
|
| 183 |
return f"https://github.com/huggingface/datasets/edit/master/datasets/{dataset_id}/README.md"
|
| 184 |
+
|
| 185 |
+
|
| 186 |
+
def create_autotrain_project_name(dataset_id: str) -> str:
|
| 187 |
+
"""Creates an AutoTrain project name for the given dataset ID."""
|
| 188 |
+
# Project names cannot have "/", so we need to format community datasets accordingly
|
| 189 |
+
dataset_id_formatted = dataset_id.replace("/", "--")
|
| 190 |
+
# Project names need to be unique, so we append a random string to guarantee this
|
| 191 |
+
project_id = str(uuid.uuid4())[:8]
|
| 192 |
+
return f"eval-project-{dataset_id_formatted}-{project_id}"
|