Update app.py

app.py

@@ -118,8 +118,8 @@ def infer(input_images, prompt, seed=42, randomize_seed=False, guidance_scale=2.
         image=concatenated_image, 
         prompt=final_prompt,
         guidance_scale=guidance_scale,
-        # width=new_width,
-        # height=new_height,
+        width=concatenated_image.size[0],
+        height=concatenated_image.size[1],
         generator=torch.Generator().manual_seed(seed),
     ).images[0]