Zen0 commited on
Commit
5e2c656
·
verified ·
1 Parent(s): 0bc7c35

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. 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"] # Changed from "text" to "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