Update BART_utils.py

BART_utils.py

@@ -7,6 +7,7 @@ import torch
 
 import requests
 import json
+from stqdm import stqdm
 
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
@@ -44,7 +45,8 @@ def get_prob(sequence, label):
 
 def get_taggs(sequence, labels, thred=0.5):
     out = []
-    for l in labels:
+    for i in stqdm(range(len(labels)):
+        l = labels[i]
         temp = get_prob(sequence, l)
         if temp >= thred:
             out.append((l, temp))