Spaces:
Running
Running
Update app.py
Browse files
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()
|