pass width, height to pipeline

app.py

@@ -32,6 +32,7 @@ def infer(control_image, seed=42, randomize_seed=False, width=1024, height=1024,
         guidance_scale=guidance_scale,
         num_inference_steps=num_inference_steps,
         generator=torch.Generator("cpu").manual_seed(seed),
+        width=width, height=height,
         **pipe_prior_output,
     ).images[0]
     return images, seed