Spaces:
Sleeping
Sleeping
Alina Lozovskaya
commited on
Commit
·
1e75b2e
1
Parent(s):
3d76e98
Fix config generation
Browse files
yourbench_space/config.py
CHANGED
@@ -18,7 +18,7 @@ def generate_base_config(
|
|
18 |
"hf_organization": hf_org,
|
19 |
"hf_dataset_name": hf_dataset_name,
|
20 |
},
|
21 |
-
"local_dataset_dir": "results/",
|
22 |
"model_list": [
|
23 |
{
|
24 |
"model_name": model_name,
|
@@ -44,6 +44,7 @@ def generate_base_config(
|
|
44 |
},
|
45 |
"upload_ingest_to_hub": {
|
46 |
"source_documents_dir": "/app/example/ingested",
|
|
|
47 |
"run": True,
|
48 |
},
|
49 |
"summarization": {"run": True},
|
@@ -83,4 +84,4 @@ def generate_base_config(
|
|
83 |
def save_config(yaml_text):
|
84 |
with open(CONFIG_PATH, "w") as file:
|
85 |
file.write(yaml_text)
|
86 |
-
return "✅ Config saved!"
|
|
|
18 |
"hf_organization": hf_org,
|
19 |
"hf_dataset_name": hf_dataset_name,
|
20 |
},
|
21 |
+
"local_dataset_dir": "/app/results/",
|
22 |
"model_list": [
|
23 |
{
|
24 |
"model_name": model_name,
|
|
|
44 |
},
|
45 |
"upload_ingest_to_hub": {
|
46 |
"source_documents_dir": "/app/example/ingested",
|
47 |
+
"hub_dataset_name": f"{hf_dataset_name}_ingested_documents",
|
48 |
"run": True,
|
49 |
},
|
50 |
"summarization": {"run": True},
|
|
|
84 |
def save_config(yaml_text):
|
85 |
with open(CONFIG_PATH, "w") as file:
|
86 |
file.write(yaml_text)
|
87 |
+
return "✅ Config saved!"
|