tqdm for debugging

app.py

@@ -12,6 +12,7 @@ from matplotlib.colors import LinearSegmentedColormap
 import matplotlib.colors as mcolors
 import plotly.express as px
 import seaborn as sns
+from tqdm import tqdm
 
 PATH = '/data/' # at least 150GB storage needs to be attached
 os.environ['TRANSFORMERS_CACHE'] = PATH
@@ -189,7 +190,7 @@ def predict_wrapper(text, language):
 
     results = []
     results_heatmap = []
-    for sentence in sentences:
+    for sentence in tqdm(sentences):
         probs = predict(sentence, model_id, tokenizer_id)
         label, probability = get_most_probable_label(probs)
         results.append([sentence, label, probability])
@@ -228,7 +229,7 @@ with gr.Blocks() as demo:
             piechart = gr.Plot()
         with gr.Column(scale=3):
             gr.Markdown(placeholder)
-            
+
     with gr.Row():
         with gr.Column(scale=7):
             plot = gr.Plot()