Update app.py

app.py

@@ -25,13 +25,10 @@ if torch.cuda.is_available():
 
 
 def pix2pix(prompt,
-    image,
-    steps,
-    image_guidance_scale,
-    width=512,
-    height=512,
-    seed
-    ):
+            image,
+            steps,
+            image_guidance_scale,,
+            seed):
     print(psutil.virtual_memory())  # print memory usage
 
     if seed == 0:
@@ -41,7 +38,7 @@ def pix2pix(prompt,
 
     #try:
     image = Image.open(image)
-    ratio = min(height / image.height, width / image.width)
+    ratio = min(512 / image.height, 512 / image.width)
     image = image.resize((int(image.width * ratio), int(image.height * ratio)), Image.LANCZOS)
 
     result = pipe(