fix: limit 3 channels in StrictTransfer

README.md

@@ -9,4 +9,4 @@ pinned: false
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
-Run with Bind Mount to autoreload when developing: docker run -v "$(pwd)":/home/user/app -p 7860:7860 farm-recolor
+Run with Bind Mount to autoreload when developing: docker build . && docker run -v "$(pwd)":/home/user/app -p 7860:7860 farm-recolor

recolorStrictTransfer.py

@@ -50,6 +50,7 @@ def color_transfer(image, old_colors_hex, new_colors_hex):
 
     # return recolored_image_path
 
-def recolor(source, old_colors, new_colors):
-    recolored = color_transfer(source, old_colors, new_colors)
+def recolor(image_np, old_colors, new_colors):
+    image_np = image_np[:, :, :3]
+    recolored = color_transfer(image_np, old_colors, new_colors)
     return recolored