Update app.py

app.py

@@ -13,14 +13,14 @@ samples = [['أنت'
                                           ,0.92],['.',0.99]]
 notes = """
 - Enter a short prompt or select (click) one of the examples and click SEND 
-- Adjust temperture (parameter) through the slider (higher usually better).
+- Adjust parameters (temperture, top k, top p and penalty) through the slider (keep close to default values).
 - Clear and enter new prompt or select another example and SEND to regenerate
 - The '.' means start a new line from no prompt (your prompt need not be long) 
 - Be patient: this runs on CPU (free tier)
 - Feedback (Twitter): @akhooli (https://twitter.com/akhooli/status/1611025232201977859)
 """
-def sayPoetry(prompt, temp=0.95):
-    gen = pipe(prompt, max_length=96, temperature = temp)[0]["generated_text"]
+def sayPoetry(prompt, temp=0.95, topk = 50, topp = 0.92, penalty=1.0):
+    gen = pipe(prompt, max_length=96, temperature = temp, top_k=topk, top_p=topp, repetition_penalty=penalty)[0]["generated_text"]
     poetry =""
     for line in gen.split('.')[:-1]:
         poetry += line + "\n"
@@ -28,7 +28,10 @@ def sayPoetry(prompt, temp=0.95):
 poetry = gr.Interface(fn=sayPoetry, 
                     inputs=[
                         gr.inputs.Textbox(label="Enter short prompt or select from examples:", placeholder="أنا الذي"), 
-                        gr.inputs.Slider(0.80, 1.0, step=0.01,default=0.95, label='control  temperature') 
+                        gr.inputs.Slider(0.80, 1.0, step=0.01,default=0.95, label='control  temperature'),
+                        gr.inputs.Slider(20, 100, step=1,default=50, label='control  top k'),
+                        gr.inputs.Slider(0.80, 1.0, step=0.01,default=1.0, label='control  top p'),
+                        gr.inputs.Slider(0.89, 1.0, step=0.01,default=1.20, label='control  penalty'),
                     ], 
                     outputs=[gr.outputs.Textbox(label="Generated Poetry:")],