import gradio as gr
from transformers import pipeline

generator = pipeline("text-generation", model="gpt2")

def generate_text(prompt):
    return generator(prompt, max_new_tokens=50)[0]["generated_text"]

gr.Interface(fn=generate_text,
             inputs=gr.Textbox(label="프롬프트 입력"),
             outputs="text",
             title="GPT-2 텍스트 생성기").launch()