U !?h@s0ddlZddlZddlmZGdddeZdS)N)BiRefNetSessionGeneralc@s(eZdZdZeddZeddZdS)BiRefNetSessionPortraitzk This class represents a BiRefNet-Portrait session, which is a subclass of BiRefNetSessionGeneral. cOsT|j||d}tjd|j||r(dnd||j||ddtj|j|||S)a% Downloads the BiRefNet-Portrait 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.onnxz^https://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-portrait-epoch_150.onnxNz$md5:c3a64a6abf20250d090cd055f12a3b67T)fnamepathZ progressbar)namepoochretrieveZchecksum_disabledZ u2net_homeosrjoin)clsargskwargsrrR/var/www/html/venv/lib/python3.8/site-packages/rembg/sessions/birefnet_portrait.pydownload_models s    z'BiRefNetSessionPortrait.download_modelscOsdS)z Returns the name of the BiRefNet-Portrait session. Parameters: *args: Additional positional arguments. **kwargs: Additional keyword arguments. Returns: str: The name of the session. zbirefnet-portraitr)r r rrrrr(s zBiRefNetSessionPortrait.nameN)__name__ __module__ __qualname____doc__ classmethodrrrrrrrs  r)r rrrrrrrs