Update app.py

app.py

@@ -79,8 +79,8 @@ def generate_image(
         apply_refiner=apply_refiner,
         api_name="/run",
     )
-    image_data = fix_base64_padding(result)  # Fix the padding of the base64 string
-    image_bytes = base64.b64decode(image_data)
+    result = fix_base64_padding(result)  # Fix the padding of the base64 string
+    image_bytes = base64.b64decode(result)
     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