Update tasks/text.py

tasks/text.py

@@ -15,7 +15,7 @@ import numpy as np
 
 router = APIRouter()
 
-DESCRIPTION = "modernBERT"
+DESCRIPTION = "modernBERT_finetuned"
 ROUTE = "/text"
 
 @router.post(ROUTE, tags=["Text Task"], 
@@ -65,7 +65,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     # Make random predictions (placeholder for actual model inference)
     true_labels = test_dataset["label"]
     # predictions = [random.randint(0, 7) for _ in range(len(true_labels))]
-    path_model = 'MatthiasPi/ModernBERT-cards-persona'
+    path_model = 'MatthiasPi/modernbert_finetunedV1'
     path_tokenizer = "answerdotai/ModernBERT-base"
     
     model = AutoModelForSequenceClassification.from_pretrained(path_model)