update model description

tasks/text.py

@@ -16,7 +16,7 @@ router = APIRouter()
 DESCRIPTION = "Random Baseline"
 ROUTE = "/text"
 
-models_description = {
+models_descriptions = {
     "baseline": "random baseline",
     "tfidf_xgb": "TF-IDF vectorizer and XGBoost classifier",
 }
@@ -49,8 +49,7 @@ def tree_classifier(test_dataset: dict, model: str):
     return predictions
 
 
-@router.post(ROUTE, tags=["Text Task"], 
-             description=DESCRIPTION)
+@router.post(ROUTE, tags=["Text Task"])
 async def evaluate_text(request: TextEvaluationRequest,
                         model: str = "tfidf_xgb"):
     """
@@ -118,7 +117,7 @@ async def evaluate_text(request: TextEvaluationRequest,
         "username": username,
         "space_url": space_url,
         "submission_timestamp": datetime.now().isoformat(),
-        "model_description": DESCRIPTION,
+        "model_description": models_descriptions[model],
         "accuracy": float(accuracy),
         "energy_consumed_wh": emissions_data.energy_consumed * 1000,
         "emissions_gco2eq": emissions_data.emissions * 1000,