Update app.py

app.py

@@ -39,6 +39,10 @@ def generate_normal_map(image):
     
     return Image.fromarray(normal_map)
 
+def fix_base64_padding(base64_str):
+    # Add padding if necessary
+    return base64_str + "=" * ((4 - len(base64_str) % 4) % 4)
+
 def generate_image(
     prompt: str,
     negative_prompt: str = "",
@@ -75,7 +79,8 @@ def generate_image(
         apply_refiner=apply_refiner,
         api_name="/run",
     )
-    image_bytes = base64.b64decode(result)
+    image_data = fix_base64_padding(result)  # Fix the padding of the base64 string
+    image_bytes = base64.b64decode(image_data)
     image = Image.open(BytesIO(image_bytes))  # Ensure the result is converted to a PIL Image
     normal_map = generate_normal_map(image)
     return image, normal_map
@@ -143,7 +148,7 @@ with gr.Blocks(css="style.css") as demo:
                 label="Height",
                 minimum=256,
                 maximum=MAX_IMAGE_SIZE,
-                step=32,
+                step32,
                 value=1024,
             )
         apply_refiner = gr.Checkbox(label="Apply refiner", value=False, visible=ENABLE_REFINER)