radius / app.py
Batnini's picture
Update app.py
e527afb verified
raw
history blame
812 Bytes
import gradio as gr
from transformers import pipeline
# Load a small Arabic-capable model (no GPU needed)
generator = pipeline('text-generation',
model='aubmindlab/aragpt2-base',
device=-1) # Force CPU
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()