install requirements automatically

游雁

2024-03-25 eb3d5c78bf764799f98ba5b19307831efd62285d

install requirements automatically

 funasr/download/download_from_hub.py

@@ -72,6 +72,9 @@
            kwargs["jieba_usr_dict"] = os.path.join(model_or_path, "jieba_usr_dict")
    if isinstance(kwargs, DictConfig):
        kwargs = OmegaConf.to_container(kwargs, resolve=True)
    if os.path.exists(os.path.join(model_or_path, "requirements.txt")):
        from funasr.utils.install_model_requirements import install_requirements
        install_requirements(os.path.join(model_or_path, "requirements.txt"))
    return kwargs

def add_file_root_path(model_or_path: str, file_path_metas: dict, cfg = {}):

 funasr/utils/install_model_requirements.py

New file
@@ -0,0 +1,35 @@
import subprocess

def install_requirements(requirements_path):
    try:
        result = subprocess.run(
            ['pip', 'install', '-r', requirements_path],
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE,
            text=True
        )

        # check status
        if result.returncode == 0:
            print("install model requirements successfully")
            return True
        else:
            print("fail to install model requirements! ")
            print("error", result.stderr)
            return False
    except Exception as e:
        result = subprocess.run(
            ['pip', 'install', '-r', requirements_path],
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE,
            text=True
        )

        # check status
        if result.returncode == 0:
            print("install model requirements successfully")
            return True
        else:
            print("fail to install model requirements! ")
            print("error", result.stderr)
            return False