akhooli commited on
Commit
35a8d0d
·
1 Parent(s): da582a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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/gpt2-ar-poetry-aub_m",
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()