Create app.py

app.py

@@ -0,0 +1,25 @@
+# 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()