app.py

import gradio as gr
from transformers import pipeline

def generate_image(text):
    dalle_mini = pipeline(‘text-to-image-generation’, model=‘CompVis/ldm-dalle-mini’)
    images = dalle_mini(text)
    return images[0][‘url’]

iface = gr.Interface(fn=generate_image,
                     inputs=gr.inputs.Textbox(lines=2, placeholder=“여기에 이미지를 생성할 텍스트를 입력하세요”),
                     outputs=“image”,
                     title=“DALL·E Mini를 이용한 이미지 생성”,
                     description=“텍스트를 입력하면, 입력된 텍스트에 기반한 이미지를 생성합니다.”)

if name == “main”:
    iface.launch()