U !?h@s0ddlZddlZddlmZGdddeZdS)N)BiRefNetSessionGeneralc@s(eZdZdZeddZeddZdS)BiRefNetSessionGeneralLitezo This class represents a BiRefNet-General-Lite session, which is a subclass of BiRefNetSessionGeneral. cOsT|j||d}tjd|j||r(dnd||j||ddtj|j|||S)a) Downloads the BiRefNet-General-Lite 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-general-bb_swin_v1_tiny-epoch_232.onnxNz$md5:4fab47adc4ff364be1713e97b7e66334T)fnamepathZ progressbar)namepoochretrieveZchecksum_disabledZ u2net_homeosrjoin)clsargskwargsrrV/var/www/html/venv/lib/python3.8/site-packages/rembg/sessions/birefnet_general_lite.pydownload_models s    z*BiRefNetSessionGeneralLite.download_modelscOsdS)a Returns the name of the BiRefNet-General-Lite session. Parameters: *args: Additional positional arguments. **kwargs: Additional keyword arguments. Returns: str: The name of the session. zbirefnet-general-liter)r r rrrrr(s zBiRefNetSessionGeneralLite.nameN)__name__ __module__ __qualname____doc__ classmethodrrrrrrrs  r)r rrrrrrrs