# app.py

import gradio as gr
from transformers import pipeline

# 모델 로드
generator = pipeline('text-to-image', model='CompVis/stable-diffusion-v1-4')

def generate_image(prompt):
    # 이미지 생성
    images = generator(prompt, num_return_sequences=1)
    return images[0]['generated_image']

# Gradio 인터페이스 설정
iface = gr.Interface(
    fn=generate_image,
    inputs=gr.inputs.Textbox(lines=2, placeholder="텍스트를 입력하세요..."),
    outputs="image",
    title="텍스트에서 이미지 생성",
    description="텍스트를 입력하면 해당 텍스트를 기반으로 이미지를 생성합니다."
)

# 인터페이스 실행
if __name__ == "__main__":
    iface.launch()