Spaces:
Runtime error
Runtime error
rodrigomasini
commited on
Commit
•
219c4e7
1
Parent(s):
05e3bde
Update app.py
Browse files
app.py
CHANGED
@@ -279,7 +279,6 @@ EXTERNAL_MODELS = [
|
|
279 |
"all-MiniLM-L6-v2",
|
280 |
"all-mpnet-base-v2",
|
281 |
"allenai-specter",
|
282 |
-
"Baichuan-text-embedding",
|
283 |
"bert-base-uncased",
|
284 |
"contriever-base-msmarco",
|
285 |
"dfm-encoder-large-v1",
|
@@ -678,17 +677,17 @@ def add_lang(examples):
|
|
678 |
|
679 |
def add_task(examples):
|
680 |
# Could be added to the dataset loading script instead
|
681 |
-
if examples["mteb_dataset_name"] in TASK_LIST_CLASSIFICATION_NORM +
|
682 |
examples["mteb_task"] = "Classification"
|
683 |
-
elif examples["mteb_dataset_name"] in TASK_LIST_CLUSTERING
|
684 |
examples["mteb_task"] = "Clustering"
|
685 |
-
elif examples["mteb_dataset_name"] in TASK_LIST_PAIR_CLASSIFICATION
|
686 |
examples["mteb_task"] = "PairClassification"
|
687 |
-
elif examples["mteb_dataset_name"] in TASK_LIST_RERANKING
|
688 |
examples["mteb_task"] = "Reranking"
|
689 |
-
elif examples["mteb_dataset_name"] in TASK_LIST_RETRIEVAL_NORM
|
690 |
examples["mteb_task"] = "Retrieval"
|
691 |
-
elif examples["mteb_dataset_name"] in TASK_LIST_STS_NORM
|
692 |
examples["mteb_task"] = "STS"
|
693 |
elif examples["mteb_dataset_name"] in TASK_LIST_SUMMARIZATION:
|
694 |
examples["mteb_task"] = "Summarization"
|
|
|
279 |
"all-MiniLM-L6-v2",
|
280 |
"all-mpnet-base-v2",
|
281 |
"allenai-specter",
|
|
|
282 |
"bert-base-uncased",
|
283 |
"contriever-base-msmarco",
|
284 |
"dfm-encoder-large-v1",
|
|
|
677 |
|
678 |
def add_task(examples):
|
679 |
# Could be added to the dataset loading script instead
|
680 |
+
if examples["mteb_dataset_name"] in TASK_LIST_CLASSIFICATION_NORM + TASK_LIST_CLASSIFICATION_SV:
|
681 |
examples["mteb_task"] = "Classification"
|
682 |
+
elif examples["mteb_dataset_name"] in TASK_LIST_CLUSTERING:
|
683 |
examples["mteb_task"] = "Clustering"
|
684 |
+
elif examples["mteb_dataset_name"] in TASK_LIST_PAIR_CLASSIFICATION:
|
685 |
examples["mteb_task"] = "PairClassification"
|
686 |
+
elif examples["mteb_dataset_name"] in TASK_LIST_RERANKING:
|
687 |
examples["mteb_task"] = "Reranking"
|
688 |
+
elif examples["mteb_dataset_name"] in TASK_LIST_RETRIEVAL_NORM:
|
689 |
examples["mteb_task"] = "Retrieval"
|
690 |
+
elif examples["mteb_dataset_name"] in TASK_LIST_STS_NORM:
|
691 |
examples["mteb_task"] = "STS"
|
692 |
elif examples["mteb_dataset_name"] in TASK_LIST_SUMMARIZATION:
|
693 |
examples["mteb_task"] = "Summarization"
|