import gradio as gr
from transformers import pipeline

# GPT 모델 불러오기
generator = pipeline("text-generation", model="gpt2")

# 사용자 입력을 받아 글 생성하는 함수
def generate_text(prompt):
    # 사용자 입력을 바탕으로 글 생성
    generated_text = generator(prompt, max_length=100)[0]['generated_text']
    return generated_text

# Gradio UI 정의
def gr_interface():
    return gr.Interface(fn=generate_text, inputs="text", outputs="text", title="한국어 텍스트 생성", description="입력된 텍스트를 바탕으로 한국어 텍스트를 생성합니다.").launch()

# Gradio 인터페이스 실행
gr_interface()