Update app.py

app.py

@@ -511,14 +511,34 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
     variant="fp16"
 ).to(device)
 
+
 def generate_image(prompt):
     # 한국어를 영어로 번역
     translated = translator(prompt)[0]['translation_text']
     
+    # Hi-res와 3840x2160 스타일 적용
+    enhanced_prompt = f"hyper-realistic 8K image of {translated}. ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic"
+    
+    # 고정된 설정값
+    negative_prompt = "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly, (deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, disgusting, amputation"
+    width = 512
+    height = 512
+    guidance_scale = 6
+    num_inference_steps = 100
+    seed = random.randint(0, 2**32 - 1)
+    generator = torch.Generator("cuda").manual_seed(seed)
+    
     # 이미지 생성
-    image = pipe(translated, num_inference_steps=30, guidance_scale=9).images[0]
+    image = pipe(
+        prompt=enhanced_prompt,
+        negative_prompt=negative_prompt,
+        width=width,
+        height=height,
+        guidance_scale=guidance_scale,
+        num_inference_steps=num_inference_steps,
+        generator=generator
+    ).images[0]
     
-    # 이미지를 PIL Image 객체로 반환
     return image
 
 def generate_images(blog_post):