from transformers import pipeline

# 변경된 코드: Stable Diffusion 모델 사용
model = pipeline('text-to-image', model='CompVis/stable-diffusion-v1-4')

# 이미지 생성 예시
generated_images = model("A cute puppy sitting on the moon")

def generate_image(prompt):
    response = model(prompt, return_tensors="pt")
    return response.images[0]

# Gradio 인터페이스 생성
iface = gr.Interface(
    fn=generate_image,
    inputs=gr.inputs.Textbox(lines=2, placeholder="여기에 텍스트를 입력하세요..."),
    outputs=gr.outputs.Image(type="pil"),
    title="텍스트를 이미지로 변환",
    description="텍스트를 입력하면 이미지를 생성합니다."
)

# 인터페이스 실행
iface.launch()