text to quote

tasks/text.py

@@ -72,7 +72,7 @@ async def evaluate_text(request: TextEvaluationRequest):
 
     try:
         # Tokenize the input texts
-        encoded_input = tokenizer(test_dataset["text"], truncation=True, padding=True, return_tensors="tf")
+        encoded_input = tokenizer(test_dataset["quote"], truncation=True, padding=True, return_tensors="tf")
         
         # Make predictions
         outputs = model(encoded_input["input_ids"], attention_mask=encoded_input["attention_mask"], training=False)