Update app.py

app.py

@@ -83,6 +83,7 @@ def sample_once(prompt, temperature=1.0, top_k=40, max_new=24):
 
 # === Generation Function ===
 def generate_names(prompt, temperature, top_k, count, retries):
+    global total_retry_count
     prompt = prompt.strip()
     if not prompt:
         raise gr.Error("Prompt cannot be empty.")