Update app.py

app.py

@@ -27,6 +27,7 @@ transformer_2=WanTransformer3DModel.from_pretrained(model_id,
 pipe = WanPipeline.from_pretrained(model_id,
                                    transformer=transformer,
                                    transformer_2 = transformer_2,
+                                   boundary_ratio = 0.3,
                                    vae=vae, torch_dtype=torch.bfloat16)
 flow_shift = 2.0
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=flow_shift)
@@ -161,7 +162,6 @@ def generate(prompt, negative_prompt, width, height, num_inference_steps, option
         num_inference_steps=num_inference_steps,
         guidance_scale=float(guidance_scale),
         guidance_scale_2=float(guidance_scale_2),
-        boundary_ratio=0.3,
     )
     image = output.frames[0][0]
     image = (image * 255).astype(np.uint8)