Update tasks/text.py

tasks/text.py

@@ -13,10 +13,12 @@ from .utils.emissions import tracker, clean_emissions_data, get_space_info
 
 router = APIRouter()
 
-MODEL_TYPE = "modernbert_frugalai"
+MODEL_TYPE = "mpnet_frugalai"
 DESCRIPTIONS = {
     "distilbert_frugalai": "distilbert tuned on frugal ai data",
-    "modernbert_frugalai": "distilbert tuned on frugal ai data"
+    "modernbert_frugalai": "distilbert tuned on frugal ai data",
+    "mpnet_frugalai": "mpnet tuned on frugal ai data",
+    
 }
 ROUTE = "/text"
 
@@ -135,7 +137,7 @@ async def evaluate_text(
     true_labels = test_dataset["label"]
     if model_type == "baseline":
         predictions = baseline_model(len(true_labels))
-    elif model_type in ["distilbert_frugalai", "modernbert_frugalai"]:
+    elif model_type in ["distilbert_frugalai", "modernbert_frugalai", "mpnet_frugalai"]:
         predictions = bert_model(test_dataset, model_type)
     else:
         raise ValueError(model_type)