Spaces:
Runtime error
Runtime error
Merge pull request #58 from huggingface/revert-57-disable-zero-shot
Browse files
app.py
CHANGED
|
@@ -43,7 +43,7 @@ TASK_TO_ID = {
|
|
| 43 |
"extractive_question_answering": 5,
|
| 44 |
"translation": 6,
|
| 45 |
"summarization": 8,
|
| 46 |
-
|
| 47 |
}
|
| 48 |
|
| 49 |
TASK_TO_DEFAULT_METRICS = {
|
|
@@ -66,7 +66,7 @@ TASK_TO_DEFAULT_METRICS = {
|
|
| 66 |
"recall",
|
| 67 |
"accuracy",
|
| 68 |
],
|
| 69 |
-
|
| 70 |
}
|
| 71 |
|
| 72 |
AUTOTRAIN_TASK_TO_LANG = {
|
|
|
|
| 43 |
"extractive_question_answering": 5,
|
| 44 |
"translation": 6,
|
| 45 |
"summarization": 8,
|
| 46 |
+
"text_zero_shot_classification": 23,
|
| 47 |
}
|
| 48 |
|
| 49 |
TASK_TO_DEFAULT_METRICS = {
|
|
|
|
| 66 |
"recall",
|
| 67 |
"accuracy",
|
| 68 |
],
|
| 69 |
+
"text_zero_shot_classification": ["accuracy", "loss"],
|
| 70 |
}
|
| 71 |
|
| 72 |
AUTOTRAIN_TASK_TO_LANG = {
|