Update app.py

app.py

@@ -5,9 +5,10 @@ from happytransformer import HappyGeneration
 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.8, max_length=150, early_stopping=False)
 
-def generate(text):
+def generate(text, length=100, temperature=0.8):
+    args_top_k = GENSettings(no_repeat_ngram_size=3, do_sample=True, top_k=80, temperature=temperature, max_length=length, early_stopping=False)
+    
     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" 
     inputText += ">" + text + "\n>"
     
@@ -30,12 +31,16 @@ examples = [
     ["be going to heaven"],
     ["be going to work"],
     ["be baking a pie"],
-    ["come home after another tiring day"]
+    ["come home after another tiring day"],
+    ["be a plague doctor"]
 ]
 
 demo = gr.Interface(
     fn=generate,
     inputs=gr.inputs.Textbox(lines=5, label="Input Text"),
+        gr.inputs.Slider(1, 200, label='Length', default=100, step=5),
+        gr.inputs.Slider(0.0, 1.0, label='Temperature - control randomness', default=0.8, step=0.1),
+        gen_button = gr.Button("Generate")
     outputs=gr.outputs.Textbox(label="Generated Text"),
     examples=examples,
     title="GPT-Greentext Playground",