Update handler.py

handler.py

@@ -6,7 +6,7 @@ from PIL import Image
 import io
 
 class CustomHandler:
-    def __init__(self, model_dir="jasperai/Flux.1-dev-Controlnet-Upscaler"):
+    def __init__(self, model_dir="huyai123/Flux.1-dev-Image-Upscaler"):
         # Load model and pipeline
         self.controlnet = FluxControlNetModel.from_pretrained(
             model_dir, torch_dtype=torch.bfloat16
@@ -54,5 +54,5 @@ class CustomHandler:
 
 # Đả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)
+    def __init__(self, model_dir="huyai123/Flux.1-dev-Image-Upscaler"):
+        super().__init__(model_dir)