app.py

import gc 
import gradio as gr
from simpletransformers.language_generation import LanguageGenerationModel
gc.collect()
# medium - was gpt2-ar-poetry-aub_m
model = LanguageGenerationModel("gpt2", "akhooli/poetry2023",
                args={"max_length": 96, "manual_seed": None, "special_tokens_list":None, "pad_token_id":0, "eos_token_id":0}, use_cuda=False)
def sayPoetry(prompt):
    gen = model.generate(prompt, verbose=False)
    poetry =""
    for line in gen[0].split('.')[:-1]:
        poetry += line + "\n"
    return poetry
                
tb = gr.Textbox(label="Enter your prompt or select from examples:", placeholder="أنا الذي", lines=1)
p_app = gr.Interface(fn=sayPoetry, inputs=tb, outputs="text", allow_flagging='never',
                   title='Arabic Poetry Generation Demo',
                   description = "a simple demo based on 1M poems fine-tuned using AraGPT2 (be patient, runs on cpu)",
                   examples=[['لولا كتاب'], ['ألا ليت'],['يا قدس'],['عيد بأية حال'], ['لكل شيئ إذا ما'],['.']])

p_app.launch()