evgeniiarazum commited on
Commit
ae2d2fc
·
verified ·
1 Parent(s): 58e2728

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +4 -4
tasks/text.py CHANGED
@@ -18,10 +18,10 @@ ROUTE = "/text"
18
 
19
  models_descriptions = {
20
  "baseline": "random baseline", # Baseline
21
- "tfidf_xgb": "TF-IDF vectorizer and XGBoost classifier", # Submitted
22
- "bert_base_pruned": "Pruned BERT base model", # Submitted
23
- 'climate_bert_pruned': "Fine-tuned and pruned DistilRoBERTa pre-trained on climate texts", # Not working
24
- "sbert_distilroberta": "Fine-tuned sentence transformer DistilRoBERTa"
25
  }
26
 
27
 
 
18
 
19
  models_descriptions = {
20
  "baseline": "random baseline", # Baseline
21
+ "distilbert_frugalai": "distilbert frugal ai",
22
+ "deberta_frugalai": "deberta frugal ai",
23
+ "modernbert_frugalai": "modernbert frugal ai",
24
+ "distilroberta_frugalai": "distilroberta frugal ai"
25
  }
26
 
27