update multi2

app.py

@@ -86,8 +86,7 @@ def infer(
     refs = [x for x in (ref1, ref2) if x is not None]
     if len(refs) == 0:
         raise gr.Error("请至少上传一张参考图（ref1 或 ref2）。")
-    if len(refs) == 2:
-        refs = [[refs[0]],[refs[1]]]
+    
 
     # 规范宽高：不超过 MAX_IMAGE_SIZE 且对齐到 16
     width = max(16, min(width, MAX_IMAGE_SIZE)) // 16 * 16
@@ -102,6 +101,9 @@ def infer(
     base_res = min(width, height, MAX_IMAGE_SIZE)
     raw_images = resize_by_bucket(refs, resolution=base_res)
 
+    if len(raw_images) == 2:
+        raw_images = [[raw_images[0]],[raw_images[1]]]
+
     # 推理
     with torch.no_grad():
         out = flux_pipeline(