imagge process

app.py

@@ -78,7 +78,7 @@ def infer(
 ):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
-
+    raw_images = [resize_by_bucket(x) for x in raw_images]
     generator = torch.Generator().manual_seed(seed)
 
     with torch.no_grad():
@@ -177,14 +177,12 @@ with gr.Blocks(css=css) as demo:
             ["a watercolor painting of yosemite valley at sunrise"],
         ]
         gr.Examples(examples=examples, inputs=[prompt])
-    raw_images=[ref1, ref2]
-    raw_images = [x for resize_by_bucket(x) in raw_images]
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=infer,
         inputs=[
             prompt,
-            raw_images,                # 新增：两张图
+            [ref1, ref2],                # 新增：两张图
             seed,
             randomize_seed,
             width,