Update tasks/text.py

tasks/text.py

@@ -64,14 +64,15 @@ async def evaluate_text(request: TextEvaluationRequest):
 
 
     # Model and Tokenizer
-    model_name = "Zen0/FrugalDisinfoHunter" 
+    # Model and Tokenizer - use the same model name for both
+    model_name = "Zen0/FrugalDisinfoHunter"
     model = AutoModelForSequenceClassification.from_pretrained(
-        "Zen0/FrugalDisinfoHunter", 
+        model_name,
         num_labels=8,
-        output_hidden_states=True,  # Additional option
-        problem_type="single_label_classification"  # Explicitly specify classification type
+        output_hidden_states=True,
+        problem_type="single_label_classification"
     )
-    tokenizer = AutoTokenizer.from_pretrained("google/mobilebert-uncased")
+    tokenizer = AutoTokenizer.from_pretrained(model_name)  # Use the same model_name
     
     # Tokenize the test data
     test_texts = test_dataset["quote"]