Update app.py

app.py

@@ -162,12 +162,32 @@ def update_dimensions_from_image(image_list):
     
     return new_width, new_height
 
-# ⚠️ spaces.GPU - duration 파라미터 없이 기본 사용
-# spaces 0.43.0에서 duration 파라미터 전달 방식이 다를 수 있음
-# progress 파라미터는 기본값 없이 사용 (gr.Progress와 @spaces.GPU 충돌 방지)
+# ⚠️ spaces.GPU 데코레이터는 파라미터 개수 제한이 있음
+# 모든 파라미터를 하나의 dict로 묶어서 전달
 @spaces.GPU
-def generate_image(prompt_embeds, image_list, width, height, num_inference_steps, guidance_scale, seed, use_turbo, progress=None):
-    prompt_embeds = prompt_embeds.to(device)
+def generate_image(generation_args):
+    """
+    generation_args = {
+        'prompt_embeds': tensor,
+        'image_list': list or None,
+        'width': int,
+        'height': int,
+        'num_inference_steps': int,
+        'guidance_scale': float,
+        'seed': int,
+        'use_turbo': bool,
+        'progress': gr.Progress or None
+    }
+    """
+    prompt_embeds = generation_args['prompt_embeds'].to(device)
+    image_list = generation_args.get('image_list')
+    width = generation_args['width']
+    height = generation_args['height']
+    num_inference_steps = generation_args['num_inference_steps']
+    guidance_scale = generation_args['guidance_scale']
+    seed = generation_args['seed']
+    use_turbo = generation_args['use_turbo']
+    progress = generation_args.get('progress')
     
     generator = torch.Generator(device=device).manual_seed(seed)
     
@@ -214,17 +234,19 @@ def infer(prompt, input_images=None, seed=42, randomize_seed=False, width=1024,
     prompt_embeds = remote_text_encoder(final_prompt)
     
     progress(0.3, desc="Waiting for GPU...")
-    image = generate_image(
-        prompt_embeds, 
-        image_list, 
-        width, 
-        height, 
-        num_inference_steps, 
-        guidance_scale, 
-        seed,
-        use_turbo,
-        progress
-    )
+    # generate_image에 dict로 전달
+    generation_args = {
+        'prompt_embeds': prompt_embeds,
+        'image_list': image_list,
+        'width': width,
+        'height': height,
+        'num_inference_steps': num_inference_steps,
+        'guidance_scale': guidance_scale,
+        'seed': seed,
+        'use_turbo': use_turbo,
+        'progress': progress
+    }
+    image = generate_image(generation_args)
     
     # 정보 로그 생성
     info_log = f"""✅ GENERATION COMPLETE!