Update tasks/text.py

tasks/text.py

@@ -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/CARDS_ModernBert_no_overfitting'
+    path_model = 'MatthiasPi/ModernBERT-cards-persona'
     path_tokenizer = "answerdotai/ModernBERT-base"
     
     model = AutoModelForSequenceClassification.from_pretrained(path_model)
@@ -75,12 +75,12 @@ async def evaluate_text(request: TextEvaluationRequest):
         return tokenizer(df["quote"], truncation=True)
     tokenized_test = test_dataset.map(preprocess_function, batched=True)
      
-    training_args = torch.load("training_args.bin")
-    training_args.eval_strategy='no'
+    # training_args = torch.load("training_args.bin")
+    # training_args.eval_strategy='no'
      
     trainer = Trainer(
         model=model,
-        args=training_args,
+        # args=training_args,
         tokenizer=tokenizer
     )