akhooli commited on
Commit
da582a3
·
1 Parent(s): 33db545

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -8,14 +8,14 @@ model = LanguageGenerationModel("gpt2", "akhooli/gpt2-ar-poetry-aub_m",
8
  def sayPoetry(prompt):
9
  gen = model.generate(prompt, verbose=False)
10
  poetry =""
11
- for line in gen[0].split('.'):
12
  poetry += line + "\n"
13
  return poetry
14
 
15
- tb = gr.Textbox(label="Enter your prompt:", placeholder="أنا الذي", lines=1)
16
  p_app = gr.Interface(fn=sayPoetry, inputs=tb, outputs="text", allow_flagging='never',
17
  title='Arabic Poetry Generation Demo',
18
- description = "a simple demo based on poems fine-tuned using AraGPT2",
19
- examples=[['لولا كتاب من'], ['ألا ليت شعري']])
20
 
21
  p_app.launch()
 
8
  def sayPoetry(prompt):
9
  gen = model.generate(prompt, verbose=False)
10
  poetry =""
11
+ for line in gen[0].split('.')[:-1]:
12
  poetry += line + "\n"
13
  return poetry
14
 
15
+ tb = gr.Textbox(label="Enter your prompt or select from examples:", placeholder="أنا الذي", lines=1)
16
  p_app = gr.Interface(fn=sayPoetry, inputs=tb, outputs="text", allow_flagging='never',
17
  title='Arabic Poetry Generation Demo',
18
+ description = "a simple demo based on poems fine-tuned using AraGPT2 (be patient, runs on cpu)",
19
+ examples=[['لولا كتاب'], ['ألا ليت'],['يا قدس'],['.']])
20
 
21
  p_app.launch()