Update app.py

app.py

@@ -116,20 +116,37 @@ def apply_mask(img: Image.Image, mask_img: Image.Image, defringe: bool = True) -
     result.paste(img, (0, 0), mask_img)
     return result
 
+
+def adjust_size_to_multiple_of_8(width: int, height: int) -> tuple[int, int]:
+    """이미지 크기를 8의 배수로 조정하는 함수"""
+    new_width = ((width + 7) // 8) * 8
+    new_height = ((height + 7) // 8) * 8
+    return new_width, new_height
+
 def generate_background(prompt: str, width: int, height: int) -> Image.Image:
     """배경 이미지 생성 함수"""
     try:
+        # 이미지 크기를 8의 배수로 조정
+        adjusted_width, adjusted_height = adjust_size_to_multiple_of_8(width, height)
+        
         with timer("Background generation"):
             image = pipe(
                 prompt=prompt,
-                width=width,
-                height=height,
+                width=adjusted_width,
+                height=adjusted_height,
                 num_inference_steps=8,
                 guidance_scale=4.0,
             ).images[0]
+            
+            # 원본 크기로 리사이즈
+            if adjusted_width != width or adjusted_height != height:
+                image = image.resize((width, height), Image.Resampling.LANCZOS)
+                
         return image
     except Exception as e:
         raise gr.Error(f"Background generation failed: {str(e)}")
+        
+
 
 def combine_with_background(foreground: Image.Image, background: Image.Image) -> Image.Image:
     """전경과 배경 합성 함수"""