Update tasks/text.py

tasks/text.py

@@ -65,7 +65,12 @@ async def evaluate_text(request: TextEvaluationRequest):
 
     # Model and Tokenizer
     model_name = "Zen0/FrugalDisinfoHunter" 
-    model = AutoModelForSequenceClassification.from_pretrained("Zen0/FrugalDisinfoHunter", num_labels=8)
+    model = AutoModelForSequenceClassification.from_pretrained(
+        "Zen0/FrugalDisinfoHunter", 
+        num_labels=8,
+        output_hidden_states=True,  # Additional option
+        problem_type="single_label_classification"  # Explicitly specify classification type
+    )
     tokenizer = AutoTokenizer.from_pretrained("google/mobilebert-uncased")
     
     # Tokenize the test data