U !?h@s0ddlZddlZddlmZGdddeZdS)N)BiRefNetSessionGeneralc@s(eZdZdZeddZeddZdS)BiRefNetSessionMassivezj This class represents a BiRefNet-Massive session, which is a subclass of BiRefNetSessionGeneral. cOsT|j||d}tjd|j||r(dnd||j||ddtj|j|||S)a$ Downloads the BiRefNet-Massive 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.onnxzmhttps://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-massive-TR_DIS5K_TR_TEs-epoch_420.onnxNz$md5:33e726a2136a3d59eb0fdf613e31e3e9T)fnamepathZ progressbar)namepoochretrieveZchecksum_disabledZ u2net_homeosrjoin)clsargskwargsrrQ/var/www/html/venv/lib/python3.8/site-packages/rembg/sessions/birefnet_massive.pydownload_models s    z&BiRefNetSessionMassive.download_modelscOsdS)z Returns the name of the BiRefNet-Massive session. Parameters: *args: Additional positional arguments. **kwargs: Additional keyword arguments. Returns: str: The name of the session. zbirefnet-massiver)r r rrrrr(s zBiRefNetSessionMassive.nameN)__name__ __module__ __qualname____doc__ classmethodrrrrrrrs  r)r rrrrrrrs