Update app.py

app.py

@@ -33,8 +33,8 @@ def generate(prompt, input_image):
         guidance_scale=30.,
         num_inference_steps=8,
         max_sequence_length=512,
-        height=image.height,
-        width=image.width,
+        height=input_image.height,
+        width=input_image.width,
         generator=torch.manual_seed(0)
     ).images[0]