Spaces:
Sleeping
Sleeping
Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
@@ -66,7 +66,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
66 |
# Model and Tokenizer
|
67 |
model_name = "Zen0/FrugalDisinfoHunter"
|
68 |
model = AutoModelForSequenceClassification.from_pretrained("Zen0/FrugalDisinfoHunter", num_labels=8)
|
69 |
-
tokenizer =
|
70 |
|
71 |
# Tokenize the test data
|
72 |
test_texts = test_dataset["quote"]
|
|
|
66 |
# Model and Tokenizer
|
67 |
model_name = "Zen0/FrugalDisinfoHunter"
|
68 |
model = AutoModelForSequenceClassification.from_pretrained("Zen0/FrugalDisinfoHunter", num_labels=8)
|
69 |
+
tokenizer = AutoTokenizer.from_pretrained("google/mobilebert-uncased")
|
70 |
|
71 |
# Tokenize the test data
|
72 |
test_texts = test_dataset["quote"]
|