src/backend/annotators/softedge_control.py

from backend.annotators.control_interface import ControlInterface
from controlnet_aux import PidiNetDetector
from PIL import Image


class SoftEdgeControl(ControlInterface):
    def get_control_image(self, image: Image) -> Image:
        processor = PidiNetDetector.from_pretrained("lllyasviel/Annotators")
        control_image = processor(image)
        return control_image