| import numpy as np | |
| from backend.annotators.control_interface import ControlInterface | |
| from controlnet_aux import LineartDetector | |
| from PIL import Image | |
| class LineArtControl(ControlInterface): | |
| def get_control_image(self, image: Image) -> Image: | |
| processor = LineartDetector.from_pretrained("lllyasviel/Annotators") | |
| control_image = processor(image) | |
| return control_image | |