U !?h@s0ddlZddlZddlmZGdddeZdS)N)BiRefNetSessionGeneralc@s(eZdZdZeddZeddZdS)BiRefNetSessionHRSODzh This class represents a BiRefNet-HRSOD session, which is a subclass of BiRefNetSessionGeneral. cOsT|j||d}tjd|j||r(dnd||j||ddtj|j|||S)a" Downloads the BiRefNet-HRSOD 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-HRSOD_DHU-epoch_115.onnxNz$md5:c017ade5de8a50ff0fd74d790d268ddaT)fnamepathZ progressbar)namepoochretrieveZchecksum_disabledZ u2net_homeosrjoin)clsargskwargsrrO/var/www/html/venv/lib/python3.8/site-packages/rembg/sessions/birefnet_hrsod.pydownload_models s    z$BiRefNetSessionHRSOD.download_modelscOsdS)z Returns the name of the BiRefNet-HRSOD session. Parameters: *args: Additional positional arguments. **kwargs: Additional keyword arguments. Returns: str: The name of the session. zbirefnet-hrsodr)r r rrrrr(s zBiRefNetSessionHRSOD.nameN)__name__ __module__ __qualname____doc__ classmethodrrrrrrrs  r)r rrrrrrrs