| from typing import Dict, List, Any | |
| from diffusers import ControlNetModel, StableDiffusionControlNetPipeline, StableDiffusionPipeline, DiffusionPipeline | |
| class EndpointHandler(): | |
| def __init__(self, path=""): | |
| self.pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0") | |
| self.pipeline.load_lora_weights(".", weight_name="/repository/safetensors.safetensors") | |
| print('hiiiii') | |
| def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]: | |
| print('data:', data) | |
| input = data['inputs'] | |
| print('input:', input) | |
| result = self.pipeline(input) | |
| print('result:', result) | |
| return result | |