Update app.py

app.py

@@ -47,6 +47,14 @@ IMAGE_WIDTH = 512
 IMAGE_HEIGHT = 812
 GUIDANCE_SCALE = 5.5
 
+def resize_to_divisible_by_8(image):
+    width, height = image.size
+    new_width = width + (8 - width % 8) if width % 8 != 0 else width
+    new_height = height + (8 - height % 8) if height % 8 != 0 else height
+    return image.resize((new_width, new_height))
+
+image = resize_to_divisible_by_8(image)
+
 # Função simples para gerar imagem
 def infer_simple(prompt):
     image = pipe(