Update app.py

app.py

@@ -57,32 +57,35 @@ MAX_IMAGE_SIZE = 1024
 
 @spaces.GPU
 def infer(prompt, 
-          ip_adapter_image = None, 
-          ip_adapter_scale = 0.5, 
-          negative_prompt = "", 
-          seed = 0, 
-          randomize_seed = False, 
-          width = 1024,  # 슬라이더 값을 기본으로 설정
-          height = 1024,  # 슬라이더 값을 기본으로 설정
-          guidance_scale = 5.0, 
-          num_inference_steps = 25
-          ):
+          ip_adapter_image=None, 
+          ip_adapter_scale=0.5, 
+          negative_prompt="", 
+          seed=0, 
+          randomize_seed=False, 
+          width=1024, 
+          height=1024, 
+          guidance_scale=5.0, 
+          num_inference_steps=25):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed)
 
+    # width와 height를 8의 배수로 조정
+    width = (width // 8) * 8
+    height = (height // 8) * 8
+
     if ip_adapter_image is None:
         pipe_t2i.to(device)
         image = pipe_t2i(
-            prompt = prompt, 
-            negative_prompt = negative_prompt,
-            guidance_scale = guidance_scale, 
-            num_inference_steps = num_inference_steps, 
-            width = width,  # 슬라이더 값을 사용
-            height = height,  # 슬라이더 값을 사용
-            generator = generator
-        ).images[0] 
-        image.save("generated_image.jpg")  # 파일 확장자를 .jpg로 변경
+            prompt=prompt, 
+            negative_prompt=negative_prompt,
+            guidance_scale=guidance_scale, 
+            num_inference_steps=num_inference_steps, 
+            width=width,  # 조정된 값을 사용
+            height=height,  # 조정된 값을 사용
+            generator=generator
+        ).images[0]
+        image.save("generated_image.jpg")
         return image, "generated_image.jpg"
     else:
         pipe_i2i.to(device)
@@ -93,14 +96,14 @@ def infer(prompt,
                 prompt=prompt,
                 ip_adapter_image=[ip_adapter_image],
                 negative_prompt=negative_prompt, 
-                height=height,  # 슬라이더 값을 사용
-                width=width,  # 슬라이더 값을 사용
+                height=height,  # 조정된 값을 사용
+                width=width,  # 조정된 값을 사용
                 num_inference_steps=num_inference_steps, 
                 guidance_scale=guidance_scale,
                 num_images_per_prompt=1,
                 generator=generator
             ).images[0]
-        image.save("generated_image.jpg")  # 파일 확장자를 .jpg로 변경
+        image.save("generated_image.jpg")
         return image, "generated_image.jpg"
 
 css="""
@@ -120,7 +123,7 @@ with gr.Blocks(css=css) as Kolors:
             with gr.Row():
                 generated_prompt = gr.Textbox(
                     label="프롬프트 입력",
-                    placeholder="이미지 생성에 사용할 프롬프트를 입력하세요",
+                    placeholder="이미지 생성에 사용할 프롬프트를 입력해 주세요.",
                     lines=2
                 )
             with gr.Row():
@@ -199,6 +202,4 @@ with gr.Blocks(css=css) as Kolors:
         outputs=[result, download_button]
     )
 
-Kolors.queue().launch(debug=True)
-
-
+Kolors.queue().launch(debug=True)