U !?h@s0ddlZddlZddlmZGdddeZdS)N)BiRefNetSessionGeneralc@s(eZdZdZeddZeddZdS)BiRefNetSessionDISzf This class represents a BiRefNet-DIS session, which is a subclass of BiRefNetSessionGeneral. cOsT|j||d}tjd|j||r(dnd||j||ddtj|j|||S)a  Downloads the BiRefNet-DIS 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-DIS-epoch_590.onnxNz$md5:2d4d44102b446f33a4ebb2e56c051f2bT)fnamepathZ progressbar)namepoochretrieveZchecksum_disabledZ u2net_homeosrjoin)clsargskwargsrrM/var/www/html/venv/lib/python3.8/site-packages/rembg/sessions/birefnet_dis.pydownload_models s    z"BiRefNetSessionDIS.download_modelscOsdS)z Returns the name of the BiRefNet-DIS session. Parameters: *args: Additional positional arguments. **kwargs: Additional keyword arguments. Returns: str: The name of the session. z birefnet-disr)r r rrrrr(s zBiRefNetSessionDIS.nameN)__name__ __module__ __qualname____doc__ classmethodrrrrrrrs  r)r rrrrrrrs