Add tqdm

app.py

@@ -8,6 +8,7 @@ import torch.nn.functional as F
 from src.gpt_base import GPT
 import json
 from huggingface_hub import hf_hub_download
+from tqdm import tqdm
 
 
 # Config class for model parameters
@@ -43,7 +44,7 @@ def decode(l):
 
 def predict_next_word(text, model, seq_len=50):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    for _ in range(seq_len):
+    for _ in tqdm(range(seq_len)):
         xb = torch.tensor(encode(text)).unsqueeze(0).to(device)
         yb = model(xb)
         next_word = yb[0, -1].argmax().item()