Update handler.py

handler.py

@@ -6,7 +6,7 @@ from PIL import Image
 import io
 
 class CustomHandler:
-    def __init__(self, model_dir="huyai123/Flux.1-dev-Image-Upscaler"):
+    def __init__(self, model_dir="jasperai/Flux.1-dev-Controlnet-Upscaler"):
         # Load model and pipeline
         self.controlnet = FluxControlNetModel.from_pretrained(
             model_dir, torch_dtype=torch.bfloat16
@@ -50,3 +50,9 @@ class CustomHandler:
         ).images[0]
         # Postprocess output
         return self.postprocess(output_image)
+
+
+# Đảm bảo rằng bạn có thể sử dụng EndpointHandler từ module handler.
+class EndpointHandler(CustomHandler):
+    def __init__(self, model_dir="jasperai/Flux.1-dev-Controlnet-Upscaler"):
+        super().__init__(model_dir)