| import gradio as gr |
| from transformers import GPT2Tokenizer, GPT2LMHeadModel |
|
|
| |
| tokenizer = GPT2Tokenizer.from_pretrained("gpt2") |
| model = GPT2LMHeadModel.from_pretrained("gpt2") |
|
|
| def generate_text(prompt): |
| inputs = tokenizer.encode(prompt, return_tensors="pt") |
| outputs = model.generate( |
| inputs, |
| max_length=100, |
| num_return_sequences=1, |
| do_sample=True, |
| temperature=0.9, |
| top_k=50, |
| top_p=0.95 |
| ) |
| return tokenizer.decode(outputs[0], skip_special_tokens=True) |
|
|
| gr.Interface( |
| fn=generate_text, |
| inputs=gr.Textbox(lines=2, label="أدخل بداية النص"), |
| outputs=gr.Textbox(label="النص المُولد"), |
| title="نموذج GPT-2", |
| description="تجربة توليد نصوص باستخدام GPT-2 الأصلي من OpenAI" |
| ).launch() |
