|
from transformers import pipeline |
|
import gradio as gr |
|
|
|
|
|
generator = pipeline('text-generation', model='gpt2') |
|
|
|
|
|
def generate_text(prompt, max_length): |
|
generated = generator(prompt, max_length=max_length, num_return_sequences=1) |
|
return generated[0]['generated_text'] |
|
|
|
|
|
interface = gr.Interface( |
|
fn=generate_text, |
|
inputs=[ |
|
gr.Textbox(lines=2, placeholder="أدخل بداية النص هنا"), |
|
gr.Slider(minimum=10, maximum=100, step=10, label="طول النص المُولد") |
|
], |
|
outputs="text", |
|
title="توليد النصوص باستخدام GPT-2", |
|
description="أدخل بداية نص وسيقوم النموذج بتوليد نص مكمل." |
|
) |
|
|
|
|
|
interface.launch() |