Update app.py

app.py

@@ -29,7 +29,7 @@ def generate(image: Image.Image, edit_prompt: str):
             share_position_embedding = True,
         )
 
-    img = image_resize(image, 320)
+    img = image_resize(image, 544)
     w, h = img.size
     img = img.resize(((w // 32) * 32, (h // 32) * 32))
     img = torch.from_numpy((np.array(img) / 127.5) - 1)