Update app.py

app.py

@@ -60,6 +60,8 @@ def infer(control_image, prompt,  image_2, prompt_2, reference_scale= 0.03 ,
     joint_attention_kwargs=dict(attention_mask=attention_mask)
     images = pipe(
         guidance_scale=guidance_scale,
+        width=width, 
+        height=height, 
         num_inference_steps=num_inference_steps,
         generator=torch.Generator("cpu").manual_seed(seed),
         joint_attention_kwargs=joint_attention_kwargs,