Update handler.py

handler.py

@@ -4,10 +4,9 @@ from io import BytesIO
 from PIL import Image
 from transparent_background import Remover
 
-
 class EndpointHandler:
-    def __init__(self):
-        # Initialize the remover model with desired settings.
+    def __init__(self, path=""):
+        self.path = path
         self.remover = Remover(mode='fast')
         # Warm up the model with a dummy image.
         dummy = Image.new("RGB", (256, 256), "white")
@@ -17,22 +16,20 @@ class EndpointHandler:
         """
         Expects a dictionary (JSON) with keys:
           - "images": a list of base64-encoded image strings (e.g. "data:image/png;base64,...")
-          - "output_type": one of "rgba", "map", "green", "blur", or "overlay"
+          - "output_type": one of "rgba", "map", "green", "blur", "overlay"
           - "threshold": a float (0.0 to 1.0)
           - "reverse": a boolean flag
         Returns a dictionary with:
           - "images": list of processed images (base64-encoded with data URI prefix)
           - "processing_times": a string with individual and total processing times
         """
-        # Get parameters from the request.
         images_data = request.get("images", [])
         output_type = request.get("output_type", "rgba")
-        threshold = request.get("threshold", 0.1)
+        threshold = request.get("threshold", 0.0)
         reverse = request.get("reverse", False)
 
         processed_results = []
         times_list = []
-
         global_start = time.time()
 
         # Process up to 3 images