Update app.py

app.py

@@ -2,10 +2,10 @@ import gradio as gr
 
 from happytransformer import HappyGeneration
 
-happy_gen = HappyGeneration("GPT-NEO", "DarwinAnim8or/GPT-Greentext-355m")
+happy_gen = HappyGeneration("GPT2", "DarwinAnim8or/GPT-Greentext-355m")
 
 from happytransformer import GENSettings
-args_top_k = GENSettings(no_repeat_ngram_size=3, do_sample=True, top_k=80, temperature=0.7, max_length=150, early_stopping=False)
+args_top_k = GENSettings(no_repeat_ngram_size=3, do_sample=True, top_k=80, temperature=0.8, max_length=150, early_stopping=False)
 
 def generate(text):
     inputText = "Write a greentext from 4chan.org. The story should be like a bullet-point list using > as the start of each line. Most greentexts are humorous or absurd in nature. Most greentexts have a twist near the end.\n" 
@@ -31,7 +31,9 @@ demo = gr.Interface(
     fn=generate,
     inputs=gr.inputs.Textbox(lines=5, label="Input Text"),
     outputs=gr.outputs.Textbox(label="Generated Text"),
-    examples=examples
+    examples=examples,
+    title="GPT-Greentext Playground",
+    description="Using the 355m size model. You may need to run it a few times in order to get something good!"
 )
 
 demo.launch()