Spaces:
Sleeping
Sleeping
Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
@@ -67,7 +67,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
67 |
model = AutoModelForSequenceClassification.from_pretrained("Zen0/FrugalDisinfoHunter", num_labels=8)
|
68 |
|
69 |
# Tokenize the test data
|
70 |
-
test_texts = test_dataset["quote"]
|
71 |
inputs = tokenizer(test_texts, padding=True, truncation=True, return_tensors="pt", max_length=512)
|
72 |
|
73 |
# Move model and inputs to GPU if available
|
|
|
67 |
model = AutoModelForSequenceClassification.from_pretrained("Zen0/FrugalDisinfoHunter", num_labels=8)
|
68 |
|
69 |
# Tokenize the test data
|
70 |
+
test_texts = test_dataset["quote"]
|
71 |
inputs = tokenizer(test_texts, padding=True, truncation=True, return_tensors="pt", max_length=512)
|
72 |
|
73 |
# Move model and inputs to GPU if available
|