Update app.py

app.py

@@ -51,7 +51,7 @@ class Theme(Base):
 
 @spaces.GPU
 def generate_parallax(image: np.ndarray, n: int):
-    input_image = Image.fromarray(image, 'RGBA')
+    input_image = Image.fromarray(image).convert('RGBA')
     detector = NudeDetector()
     layers = []
     indexes = []
@@ -77,7 +77,7 @@ def generate_parallax(image: np.ndarray, n: int):
 
 @spaces.GPU
 def generate_animation(image: np.ndarray):
-    return generate_animation_images(Image.fromarray(image, 'RGBA'))
+    return generate_animation_images(Image.fromarray(image).convert('RGBA'))
 
 
 with gr.Blocks(theme=Theme()) as demo: