Update tasks/text.py

tasks/text.py

@@ -72,9 +72,9 @@ async def evaluate_text(request: TextEvaluationRequest):
     predictions = []
     
     for i in range(len(test_dataset["quote"])):
-    encoded_input = tokenizer(test_dataset["quote"][i], truncation=True, padding=True, return_tensors="tf")
-    outputs = model(encoded_input["input_ids"], attention_mask=encoded_input["attention_mask"], training=False)
-    predictions.append(tf.argmax(outputs.logits, axis=1))
+        encoded_input = tokenizer(test_dataset["quote"][i], truncation=True, padding=True, return_tensors="tf")
+        outputs = model(encoded_input["input_ids"], attention_mask=encoded_input["attention_mask"], training=False)
+        predictions.append(tf.argmax(outputs.logits, axis=1))
     
     # Get true labels
     true_labels = test_dataset["label"]