Update app.py

app.py

@@ -20,6 +20,7 @@ SPECIAL = ['<pad>', '<bos>', '<eos>', '<sep>']
 PAD, BOS, EOS, SEP = [stoi[s] for s in SPECIAL]
 VOCAB_SIZE = len(itos)
 MAX_LEN = 128
+total_retry_count = 0
 
 # === Model ===
 class GPTSmall(nn.Module):
@@ -91,6 +92,8 @@ def generate_names(prompt, temperature, top_k, count, retries):
     results = []
     for _ in range(count):
         for attempt in range(retries):
+            print("Retrying generation...")
+            total_retry_count++
             name = sample_once(prompt, temperature=temperature, top_k=top_k)
             if len(name) >= 4:
                 results.append({"Generated Name": name})
@@ -98,6 +101,7 @@ def generate_names(prompt, temperature, top_k, count, retries):
     df = pd.DataFrame(results)
     file_path = tempfile.NamedTemporaryFile(delete=False, suffix=".txt").name
     df.to_csv(file_path, index=False, header=False)
+    print(f"Total retries: {total_retry_count}")
     return df, file_path
 
 # === UI ===