from typing import Dict, List, Any | |
from diffusers import ControlNetModel, StableDiffusionControlNetPipeline, StableDiffusionPipeline | |
class EndpointHandler(): | |
def __init__(self, path=""): | |
self.pipeline = StableDiffusionPipeline.from_single_file( | |
"https://huggingface.co/artificialguybr/LogoRedmond-LogoLoraForSDXL/blob/main/LogoRedmond_LogoRedAF.safetensors" | |
) | |
def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]: | |
""" | |
data args: | |
inputs (:obj: `str` | `PIL.Image` | `np.array`) | |
kwargs | |
Return: | |
A :obj:`list` | `dict`: will be serialized and returned | |
""" | |
# Use self.pipeline for inference | |
# result = self.pipeline(your_input_data) | |
# return result | |