# app.py
import gradio as gr
from transformers import pipeline

# 텍스트 생성 파이프라인 로드
generator = pipeline('text-generation', model='skt/ko-gpt-trinity-1.2B-v0.5')

# 생성 함수 정의
def generate_text(prompt, max_length=100):
    results = generator(prompt, max_length=max_length, num_return_sequences=1)
    return results[0]['generated_text']

# Gradio 인터페이스 정의
demo = gr.Interface(
    fn=generate_text,
    inputs=[
        gr.Textbox(placeholder="텍스트를 입력하세요...", label="프롬프트"),
        gr.Slider(minimum=10, maximum=200, value=100, label="최대 길이")
    ],
    outputs=gr.Textbox(label="생성된 텍스트")
)

# 앱 실행
demo.launch()