Update app.py

app.py

@@ -1,8 +1,9 @@
 import gradio as gr
 import torch
 import torch.nn as nn
-import re, unicodedata, random, math
+import re, unicodedata, random
 from pathlib import Path
+import pandas as pd
 
 # === Constants and Config ===
 DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -72,7 +73,8 @@ def sample_once(prompt, temperature=1.0, top_k=40, max_new=24):
             break
         seq.append(idx)
 
-    name = ''.join(itos[i] for i in seq if i not in {BOS, SEP, EOS, PAD})
+    generated = [itos[i] for i in seq if i not in {BOS, SEP, EOS, PAD}]
+    name = ''.join(generated).replace(prompt, "").strip()
     return clean_name(name)
 
 # === Gradio UI ===
@@ -80,8 +82,8 @@ def generate_ui(prompt, temperature, top_k, count):
     results = []
     for _ in range(count):
         name = sample_once(prompt, temperature=temperature, top_k=top_k)
-        results.append(name)
-    return "\n".join(results)
+        results.append({"Generated Name": name})
+    return pd.DataFrame(results)
 
 description = """🎭 **Fantasy Name Generator**  
 Give it a prompt like `a forgotten warrior king` or `mistress of the black swamp` and it'll generate creative fantasy-style names.  
@@ -96,7 +98,7 @@ with gr.Blocks() as demo:
         top_k = gr.Slider(10, 100, step=10, value=40, label="Top-K")
         count = gr.Slider(1, 5, step=1, value=3, label="Names to Generate")
     generate_btn = gr.Button("Generate Names")
-    output = gr.Textbox(label="Generated Names", lines=5)
+    output = gr.Dataframe(headers=["Generated Name"], datatype="str", label="Generated Fantasy Names", interactive=False)
     generate_btn.click(fn=generate_ui, inputs=[prompt, temperature, top_k, count], outputs=output)
 
 demo.launch()