Update handle to convert colors

handler.py

@@ -21,8 +21,10 @@ class EndpointHandler():
         inputs = data.pop("inputs", data)
         # parameters = data.pop("parameters", {"mode": "image"})
 
-        image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
-        trimap = Image.open(BytesIO(base64.b64decode(inputs['trimap'])))
+        image = Image.open(
+            BytesIO(base64.b64decode(inputs['image']))).convert("RGB")
+        trimap = Image.open(
+            BytesIO(base64.b64decode(inputs['trimap']))).convert("L")
         # image = data.pop("image")
         # trimap = data.pop("trimap")
 

requirements.txt

@@ -1,12 +0,0 @@
-uvicorn==0.23.2
-fastapi==0.104.1
-python-multipart==0.0.6
-pydantic==2.4.2
-
-Pillow==9.5.0
-numpy==1.24.3
-torch==2.0.*
-torchvision
-transformers==4.37.2
-
-segment-anything @ git+https://github.com/facebookresearch/segment-anything.git@6fdee8f2727f4506cfbbe553e23b895e27956588