Update app.py

app.py

@@ -61,7 +61,7 @@ def clean_name(text, title_case=True, max_repeats=2):
     text = re.sub(r'\b(The|Of|In|On|A)\s+\1\b', r'\1', text, flags=re.IGNORECASE)
     return re.sub(r"([a-zA-Z])'S\b", lambda m: m.group(1) + "'s", text)
 
-def sample_once(prompt, temperature=1.0, top_k=40, max_new=24):
+def sample_once(prompt, temperature=1.0, top_k=40, max_new=40):
     sample_i = time.time()
     seq = [BOS] + [stoi.get(c, PAD) for c in prompt] + [SEP]
     for _ in range(max_new):
@@ -96,7 +96,7 @@ def generate_names(prompt, temperature, top_k, count, retries):
             print("Retrying generation...")
             total_retry_count = total_retry_count + 1
             name = sample_once(prompt, temperature=temperature, top_k=top_k)
-            if len(name) >= 4:
+            if len(name) >= 3:
                 results.append({"Generated Name": name})
                 break
     df = pd.DataFrame(results)