File size: 405 Bytes
4b33f36
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
"""HF Inference Endpoints entry point, same base64 response as the SDXL repo."""
from flux3_engine import Engine


class EndpointHandler:
    def __init__(self, path=''):
        self.engine = Engine(model_path=path or None)
        self.engine.load()

    def __call__(self, data):
        # HF and the existing persona caller send the legacy SDXL-shaped payload.
        return self.engine.legacy(data)