Update tasks/text.py

tasks/text.py

@@ -61,10 +61,10 @@ async def evaluate_text(request: TextEvaluationRequest):
     #--------------------------------------------------------------------------------------------   
 
 
-        # Model and Tokenizer
-    model_name = "Zen0/FrugalDisinfoHunter"
+    # Model and Tokenizer
+    model_name = "bert-base-uncased"  # Use the base model architecture
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForSequenceClassification.from_pretrained(model_name)
+    model = AutoModelForSequenceClassification.from_pretrained("Zen0/FrugalDisinfoHunter", num_labels=8)
     
     # Tokenize the test data
     test_texts = test_dataset["quote"]  # Changed from "text" to "quote"