Spaces:
Runtime error
Runtime error
Add translation
Browse files
app.py
CHANGED
|
@@ -33,7 +33,7 @@ TASK_TO_ID = {
|
|
| 33 |
# "multi_label_classification": 3, # Not fully supported in AutoTrain
|
| 34 |
"entity_extraction": 4,
|
| 35 |
"extractive_question_answering": 5,
|
| 36 |
-
|
| 37 |
"summarization": 8,
|
| 38 |
}
|
| 39 |
|
|
@@ -276,10 +276,12 @@ with st.form(key="form"):
|
|
| 276 |
project_id = str(uuid.uuid4())[:3]
|
| 277 |
payload = {
|
| 278 |
"username": AUTOTRAIN_USERNAME,
|
| 279 |
-
"proj_name": f"
|
| 280 |
"task": TASK_TO_ID[selected_task],
|
| 281 |
"config": {
|
| 282 |
-
"language": "en"
|
|
|
|
|
|
|
| 283 |
"max_models": 5,
|
| 284 |
"instance": {
|
| 285 |
"provider": "aws",
|
|
|
|
| 33 |
# "multi_label_classification": 3, # Not fully supported in AutoTrain
|
| 34 |
"entity_extraction": 4,
|
| 35 |
"extractive_question_answering": 5,
|
| 36 |
+
"translation": 6,
|
| 37 |
"summarization": 8,
|
| 38 |
}
|
| 39 |
|
|
|
|
| 276 |
project_id = str(uuid.uuid4())[:3]
|
| 277 |
payload = {
|
| 278 |
"username": AUTOTRAIN_USERNAME,
|
| 279 |
+
"proj_name": f"eval-project-{project_id}",
|
| 280 |
"task": TASK_TO_ID[selected_task],
|
| 281 |
"config": {
|
| 282 |
+
"language": "en"
|
| 283 |
+
if selected_task != "translation"
|
| 284 |
+
else "en2de", # Need this dummy pair to enable translation
|
| 285 |
"max_models": 5,
|
| 286 |
"instance": {
|
| 287 |
"provider": "aws",
|