overlay revised2

app.py

@@ -365,7 +365,7 @@ with gr.Blocks(
                 return segmented_image_pil
             
             # Overlay logic
-            seg_rgba_np = np.array(segmented_image_pil.convert("RGBA"))
+            seg_rgba_np = np.array(segmented_image_pil.convert("RGB"))
             input_L_np = np.array(original_lineart_pil.convert("L"))
             
             # Normalize input_L_np to 0-1 and expand dims for broadcasting
@@ -373,7 +373,7 @@ with gr.Blocks(
             
             # Modulate all channels of segmentation_rgba by the grayscale factor
             overlay_np = (seg_rgba_np * factor).astype(np.uint8)
-            disp_segmented = Image.fromarray(overlay_np, 'RGBA')
+            disp_segmented = Image.fromarray(overlay_np, 'RGB')
             
             return disp_segmented
         else: