| import gradio as gr |
| from transformers import pipeline |
|
|
| |
| generator = pipeline('text-generation', |
| model='aubmindlab/aragpt2-base', |
| device=-1) |
|
|
| def generate_arabic_text(prompt): |
| output = generator( |
| prompt, |
| max_length=100, |
| num_return_sequences=1, |
| pad_token_id=generator.tokenizer.eos_token_id |
| ) |
| return output[0]['generated_text'] |
|
|
| with gr.Blocks() as demo: |
| gr.Markdown("## مولد نصوص عربية صغير") |
| with gr.Row(): |
| inp = gr.Textbox(label="اكتب مقدمة للنص") |
| btn = gr.Button("ولّد") |
| out = gr.Textbox(label="النص المولد") |
| |
| btn.click(generate_arabic_text, inputs=inp, outputs=out) |
|
|
| demo.launch() |