Include ngram size penalty as slider

app.py

@@ -6,10 +6,10 @@ happy_gen = HappyGeneration("GPT2", "DarwinAnim8or/GPT-Greentext-355m")
 
 from happytransformer import GENSettings
 
-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)
+def generate(text, length=100, penalty=3, temperature=0.8):
+    args_top_k = GENSettings(no_repeat_ngram_size=penalty, 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 = "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>"
     
     result = happy_gen.generate_text(inputText, args=args_top_k)
@@ -37,9 +37,12 @@ examples = [
 
 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)],
+    inputs=[
+        gr.inputs.Textbox(lines=5, label="Input Text"),
+        gr.inputs.Slider(5, 200, label='Length', default=100, step=5),
+        gr.inputs.Slider(1, 10, label='Repetition penalty', default=3, step=1),
+        gr.inputs.Slider(0.0, 1.0, label='Temperature - control randomness', default=0.8, step=0.1)
+    ],
     outputs=gr.outputs.Textbox(label="Generated Text"),
     examples=examples,
     title="GPT-Greentext Playground",