Spaces:
Running
on
Zero
KeyError: 'HOME'
Hello brother, after following your tutorial #1 to deploy, I got this error? Do I need to download the model file for ZhengPeng7\BiRefNet? Please help me take a look at how to solve this issue, thank you very much!
Error Message:
Traceback (most recent call last):
File "F:\AI\locally-compatible-BG-removal\app.py", line 11, in
birefnet = AutoModelForImageSegmentation.from_pretrained(
File "D:\Miniconda\envs\background-removal\lib\site-packages\transformers\models\auto\auto_factory.py", line 551, in from_pretrained
model_class = get_class_from_dynamic_module(
File "D:\Miniconda\envs\background-removal\lib\site-packages\transformers\dynamic_module_utils.py", line 514, in get_class_from_dynamic_module
return get_class_in_module(class_name, final_module)
File "D:\Miniconda\envs\background-removal\lib\site-packages\transformers\dynamic_module_utils.py", line 212, in get_class_in_module
module_spec.loader.exec_module(module)
File "", line 883, in exec_module
File "", line 241, in _call_with_frames_removed
File "C:\Users\HCl.cache\huggingface\modules\transformers_modules\ZhengPeng7\BiRefNet\99ec97615c3b4c61302d6e798bab9e991c564cde\birefnet.py", line 1392, in
config = Config()
File "C:\Users\HCl.cache\huggingface\modules\transformers_modules\ZhengPeng7\BiRefNet\99ec97615c3b4c61302d6e798bab9e991c564cde\birefnet.py", line 10, in init
self.sys_home_dir = os.environ['HOME'] # Make up your file system as: SYS_HOME_DIR/codes/dis/BiRefNet, SYS_HOME_DIR/datasets/dis/xx, SYS_HOME_DIR/weights/xx
File "D:\Miniconda\envs\background-removal\lib\os.py", line 679, in getitem
raise KeyError(key) from None
KeyError: 'HOME'
@walkingwithGod
awesome catch 👏
I will try to provide a work around tomorrow.
on another note friendly tagging @ZhengPeng7 : the error stems from https://huggingface.co/ZhengPeng7/BiRefNet/blob/main/birefnet.py#L10 since Windows does not have a "HOME" .
Thank you very much!
Yeah, you can refer to the comments in that line for the right modification. But for only inference, you can just assign any str to it, such as TMP
, XXX
, etc.
Yeah, you can refer to the comments in that line for the right modification. But for only inference, you can just assign any str to it, such as
TMP
,XXX
, etc.
OK,Thank you!
Hi
@walkingwithGod
I have pushed a fix on the original model, try rerunning the script again, and it should work without any changes.
Let me know if you encounter any future errors
Hi @walkingwithGod
I have pushed a fix on the original model, try rerunning the script again, and it should work without any changes.
Let me know if you encounter any future errors
Thank you very much!