Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -2,8 +2,8 @@ import gc
|
|
2 |
import gradio as gr
|
3 |
from simpletransformers.language_generation import LanguageGenerationModel
|
4 |
gc.collect()
|
5 |
-
# medium
|
6 |
-
model = LanguageGenerationModel("gpt2", "akhooli/
|
7 |
args={"max_length": 96, "manual_seed": None, "special_tokens_list":None, "pad_token_id":0, "eos_token_id":0}, use_cuda=False)
|
8 |
def sayPoetry(prompt):
|
9 |
gen = model.generate(prompt, verbose=False)
|
@@ -16,6 +16,6 @@ tb = gr.Textbox(label="Enter your prompt or select from examples:", placeholder=
|
|
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()
|
|
|
2 |
import gradio as gr
|
3 |
from simpletransformers.language_generation import LanguageGenerationModel
|
4 |
gc.collect()
|
5 |
+
# medium - was gpt2-ar-poetry-aub_m
|
6 |
+
model = LanguageGenerationModel("gpt2", "akhooli/poetry2023",
|
7 |
args={"max_length": 96, "manual_seed": None, "special_tokens_list":None, "pad_token_id":0, "eos_token_id":0}, use_cuda=False)
|
8 |
def sayPoetry(prompt):
|
9 |
gen = model.generate(prompt, verbose=False)
|
|
|
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()
|