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