rodrigomasini commited on
Commit
219c4e7
1 Parent(s): 05e3bde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
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 + TASK_LIST_CLASSIFICATION_DA + TASK_LIST_CLASSIFICATION_NB + TASK_LIST_CLASSIFICATION_PL + TASK_LIST_CLASSIFICATION_SV + TASK_LIST_CLASSIFICATION_ZH:
682
  examples["mteb_task"] = "Classification"
683
- elif examples["mteb_dataset_name"] in TASK_LIST_CLUSTERING + TASK_LIST_CLUSTERING_DE + TASK_LIST_CLUSTERING_PL + TASK_LIST_CLUSTERING_ZH:
684
  examples["mteb_task"] = "Clustering"
685
- elif examples["mteb_dataset_name"] in TASK_LIST_PAIR_CLASSIFICATION + TASK_LIST_PAIR_CLASSIFICATION_PL + TASK_LIST_PAIR_CLASSIFICATION_ZH:
686
  examples["mteb_task"] = "PairClassification"
687
- elif examples["mteb_dataset_name"] in TASK_LIST_RERANKING + TASK_LIST_RERANKING_ZH:
688
  examples["mteb_task"] = "Reranking"
689
- elif examples["mteb_dataset_name"] in TASK_LIST_RETRIEVAL_NORM + TASK_LIST_RETRIEVAL_PL + TASK_LIST_RETRIEVAL_ZH:
690
  examples["mteb_task"] = "Retrieval"
691
- elif examples["mteb_dataset_name"] in TASK_LIST_STS_NORM + TASK_LIST_STS_PL + TASK_LIST_STS_ZH:
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"