Update app.py

app.py

@@ -1,8 +1,10 @@
-import gradio as gr
 from transformers import pipeline
 
-# Hugging Face 모델 파이프라인 로드
-model = pipeline('image-generation', model='CompVis/dalle-mini')
+# 변경된 코드: 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")