Add installation of packages required to run lama-cleaner

main
Uminosachi 2024-04-01 16:46:52 +09:00
parent 6305c6189e
commit e2725802be
1 changed files with 17 additions and 5 deletions

View File

@ -48,11 +48,23 @@ if not launch.is_installed("transformers"):
except Exception:
print("Can't install transformers. Please follow the readme to install manually")
# if not launch.is_installed("lama_cleaner"):
# try:
# launch.run_pip("install lama-cleaner", "requirements for lama_cleaner")
# except Exception:
# print("Can't install lama-cleaner. Please follow the readme to install manually")
if not launch.is_installed("loguru"):
try:
launch.run_pip("install loguru", "requirements for loguru")
except Exception:
print("Can't install loguru. Please follow the readme to install manually")
if not launch.is_installed("rich"):
try:
launch.run_pip("install rich", "requirements for rich")
except Exception:
print("Can't install rich. Please follow the readme to install manually")
if not launch.is_installed("pydantic"):
try:
launch.run_pip("install pydantic", "requirements for pydantic")
except Exception:
print("Can't install pydantic. Please follow the readme to install manually")
if not launch.is_installed("ultralytics"):
try: