Update app.py

app.py

@@ -40,8 +40,7 @@ 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)
+    return base64_str + "=" * (-len(base64_str) % 4)
 
 def generate_image(
     prompt: str,
@@ -80,7 +79,10 @@ def generate_image(
         api_name="/run",
     )
     result = fix_base64_padding(result)  # Fix the padding of the base64 string
-    image_bytes = base64.b64decode(result)
+    try:
+        image_bytes = base64.b64decode(result)
+    except binascii.Error:
+        raise ValueError("Error decoding the Base64 string. Please check the input.")
     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