akhooli commited on
Commit
646c6c0
1 Parent(s): 8a73eeb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -6
app.py CHANGED
@@ -19,7 +19,8 @@ notes = """
19
  - Clear and enter new prompt or select another example and SEND to regenerate
20
  - The '.' means start a new line from no prompt (your prompt need not be long)
21
  - Be patient: this runs on CPU (free tier)
22
- - Feedback (Twitter): @akhooli (https://twitter.com/akhooli/status/1611025232201977859)
 
23
  """
24
  def sayPoetry(prompt, temp=0.95, topk = 50, topp = 0.92, penalty=1.0):
25
  gen = pipe(prompt, max_length=96, temperature = temp, top_k=topk, top_p=topp, repetition_penalty=penalty)[0]["generated_text"]
@@ -35,11 +36,12 @@ poetry = gr.Interface(fn=sayPoetry,
35
  gr.inputs.Slider(0.80, 1.0, step=0.01,default=1.0, label='control top p'),
36
  gr.inputs.Slider(0.90, 1.20, step=0.01,default=1.20, label='control penalty'),
37
  ],
38
- outputs=[gr.outputs.Textbox(label="Generated Poetry:")],
39
 
40
- allow_flagging='never',
41
- title='Arabic Poetry Generation Demo (updated Jan. 2023)',
42
- description = "A simple demo of AI generated poetry based on 1M poems fine-tuned using AraGPT2 (be patient, runs on cpu)",
43
- examples=samples,
 
44
  article = notes)
45
  poetry.launch()
 
19
  - Clear and enter new prompt or select another example and SEND to regenerate
20
  - The '.' means start a new line from no prompt (your prompt need not be long)
21
  - Be patient: this runs on CPU (free tier)
22
+ - Feedback (Twitter): @akhooli (https://twitter.com/akhooli/status/1611025232201977859)
23
+ - Note/Disclaimer: may generate unaccepted or inappropriate contednt. Use at your own risk.
24
  """
25
  def sayPoetry(prompt, temp=0.95, topk = 50, topp = 0.92, penalty=1.0):
26
  gen = pipe(prompt, max_length=96, temperature = temp, top_k=topk, top_p=topp, repetition_penalty=penalty)[0]["generated_text"]
 
36
  gr.inputs.Slider(0.80, 1.0, step=0.01,default=1.0, label='control top p'),
37
  gr.inputs.Slider(0.90, 1.20, step=0.01,default=1.20, label='control penalty'),
38
  ],
39
+ outputs=[gr.outputs.Textbox(label="Generated Poetry:")],
40
 
41
+ allow_flagging='never',
42
+ title='Arabic Poetry Generation Demo (updated Jan. 2023)',
43
+ description = "A simple demo of AI generated poetry based on 1M poems fine-tuned using AraGPT2 (be patient, runs on cpu)",
44
+ examples=samples,
45
+ cache_examples=False,
46
  article = notes)
47
  poetry.launch()