Update app.py

app.py

@@ -29,6 +29,8 @@ def pix2pix(
     image,
     steps,
     image_guidance_scale,
+    width=512,
+    height=512,
     seed
 ):
     print(psutil.virtual_memory())  # print memory usage