Update app.py

app.py

@@ -78,6 +78,7 @@
 # )
 
 # demo.launch()
+
 import sentencepiece as spm
 import numpy as np
 import tensorflow as tf
@@ -138,7 +139,7 @@ def generate_text(seed_text, next_words=30, temperature=0.5):
         gr.Warning("Flagged potentially harmful output.")
         decoded = 'Flagged Output'
 
-    return {"text": decoded, "flag_button": True}
+    return decoded
 
 flagged_outputs = []
 
@@ -177,10 +178,11 @@ demo = gr.Interface(
         gr.Slider(0.1, 1, value=0.5, label='Temperature', info='Controls randomness of generation, higher values = more creative, lower values = more probabilistic')
     ],
     outputs=[
-        gr.JSON(label="Generated Names with Flag Button")
+        gr.Textbox(label="Generated Name"),
+        gr.Button("Flag Output")
     ],
     title='Dungen Dev - Name Generator',
-    description=None,
+    description='A prompt-based name generator for game developers. Dungen Dev is an experimental model, and may produce outputs that are inappropriate, biased, or potentially harmful and inaccurate. Caution is advised.',
     examples=[
         ["a male character name", 30, 0.5],
         ["a futuristic city name", 30, 0.5],