Update Highlighter.py

Highlighter.py

@@ -62,7 +62,7 @@ class Highlighter():
         dl = DataLoader(ds, batch_size=4)
 
         res = []
-        for data in tqdm(dl):
+        for data in dl:
             ids, am = data["input_ids"], data["attention_mask"]
             
             out = self.model(input_ids=ids.to(self.device), attention_mask=am.to(self.device))