Update tasks/text.py

tasks/text.py

@@ -5,6 +5,7 @@ from sklearn.metrics import accuracy_score
 import random
 import numpy as np
 from huggingface_hub import PyTorchModelHubMixin
+from tqdm import tqdm, trange
 
 import torch
 import torch.nn as nn
@@ -106,7 +107,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     test_dataloader = DataLoader(test_data, sampler=test_sampler, batch_size=batch_size)
 
     predictions = []
-    for step, batch in enumerate(test_dataloader):
+    for batch in tqdm(test_dataloader):
 
 
         b_input_ids, b_input_mask, b_token_type_ids = batch