U !?h@s0ddlZddlZddlmZGdddeZdS)N)BiRefNetSessionGeneralc@s(eZdZdZeddZeddZdS)BiRefNetSessionCODzf This class represents a BiRefNet-COD session, which is a subclass of BiRefNetSessionGeneral. cOsT|j||d}tjd|j||r(dnd||j||ddtj|j|||S)a  Downloads the BiRefNet-COD model file from a specific URL and saves it. Parameters: *args: Additional positional arguments. **kwargs: Additional keyword arguments. Returns: str: The path to the downloaded model file. z.onnxzYhttps://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-COD-epoch_125.onnxNz$md5:f6d0d21ca89d287f17e7afe9f5fd3b45T)fnamepathZ progressbar)namepoochretrieveZchecksum_disabledZ u2net_homeosrjoin)clsargskwargsrrM/var/www/html/venv/lib/python3.8/site-packages/rembg/sessions/birefnet_cod.pydownload_models s    z"BiRefNetSessionCOD.download_modelscOsdS)z Returns the name of the BiRefNet-COD session. Parameters: *args: Additional positional arguments. **kwargs: Additional keyword arguments. Returns: str: The name of the session. z birefnet-codr)r r rrrrr(s zBiRefNetSessionCOD.nameN)__name__ __module__ __qualname____doc__ classmethodrrrrrrrs  r)r rrrrrrrs