Update app.py

app.py

@@ -511,7 +511,15 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
     variant="fp16"
 ).to(device)
 
-
+def generate_image(prompt):
+    # 한국어를 영어로 번역
+    translated = translator(prompt)[0]['translation_text']
+    
+    # 이미지 생성
+    image = pipe(translated, num_inference_steps=30, guidance_scale=9).images[0]
+    
+    # 이미지를 PIL Image 객체로 반환
+    return image
 
 def generate_images(blog_post):
     images = []