a
    Aþdî  ã                   @   s&   d dl mZ G dd„ dƒZdd„ ZdS )é    )Úsharedc                   @   s   e Zd Zdd„ Zdd„ ZdS )ÚFaceRestorationc                 C   s   dS )NÚNone© )Úselfr   r   ú@/var/www/html/stable-diffusion-webui/modules/face_restoration.pyÚname   s    zFaceRestoration.namec                 C   s   |S ©Nr   )r   Únp_imager   r   r   Úrestore   s    zFaceRestoration.restoreN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r      s   r   c                 C   s2   dd„ t jD ƒ}t|ƒdkr | S |d }| | ¡S )Nc                 S   s,   g | ]$}|  ¡ tjjks$tjjd u r|‘qS r	   )r   r   ÚoptsÚface_restoration_model)Ú.0Úxr   r   r   Ú
<listcomp>   ó    z!restore_faces.<locals>.<listcomp>r   )r   Úface_restorersÚlenr   )r
   r   Zface_restorerr   r   r   Úrestore_faces   s
    r   N)Úmodulesr   r   r   r   r   r   r   Ú<module>   s   