Update handler.py
Browse files- handler.py +0 -6
handler.py
CHANGED
@@ -50,9 +50,3 @@ class CustomHandler:
|
|
50 |
).images[0]
|
51 |
# Postprocess output
|
52 |
return self.postprocess(output_image)
|
53 |
-
|
54 |
-
|
55 |
-
# Đảm bảo rằng bạn có thể sử dụng EndpointHandler từ module handler.
|
56 |
-
class EndpointHandler(CustomHandler):
|
57 |
-
def __init__(self, model_dir="huyai123/Flux.1-dev-Image-Upscaler"):
|
58 |
-
super().__init__(model_dir)
|
|
|
50 |
).images[0]
|
51 |
# Postprocess output
|
52 |
return self.postprocess(output_image)
|
|
|
|
|
|
|
|
|
|
|
|