search

danielgatis / rembg

Rembg is a tool to remove images background
MIT License
16.48k stars 1.84k forks source link

[BUG] Fine-tuned model is always overwritten #443

Closed micko93ns closed 1 year ago

micko93ns commented 1 year ago

Fine-tuned model is always overwritten by the one from github repo.

pooch.retrieve( url, f"md5:{md5}", fname=fname, path=Path(u2net_home).expanduser(), progressbar=True, )

Can you add additional parameter to remove function to avoid this behaviour? Or do you have another suggest how to use fine-tuned model with rembg?

Thanks for a great work!

danielgatis commented 1 year ago

try to set the env var: 

MODEL_CHECKSUM_DISABLED=1