diff --git a/README.md b/README.md index 5c09ebd..30fff4e 100644 --- a/README.md +++ b/README.md @@ -61,6 +61,14 @@ venv\Scripts\activate python app.py ``` +## Update + +```bash +git pull +venv\Scripts\activate +pip install -r requirements.txt --upgrade +``` + ## Troubleshooting ``` diff --git a/app.py b/app.py index c711387..d7add9a 100644 --- a/app.py +++ b/app.py @@ -42,11 +42,6 @@ models = [ if len(models) == 0: raise ValueError("No model found in `weights` folder") models.sort() -hubert_model = None - -print("Loading rmvpe model...") -rmvpe_model = RMVPE("rmvpe.pt", config.is_half, config.device) -print("rmvpe model loaded.") def model_data(model_name): @@ -115,7 +110,16 @@ def load_hubert(): hubert_model = hubert_model.half() else: hubert_model = hubert_model.float() - hubert_model.eval() + return hubert_model.eval() + + +print("Loading hubert model...") +hubert_model = load_hubert() +print("Hubert model loaded.") + +print("Loading rmvpe model...") +rmvpe_model = RMVPE("rmvpe.pt", config.is_half, config.device) +print("rmvpe model loaded.") def tts( diff --git a/requirements.txt b/requirements.txt index 2ac9e88..3bba8c2 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,10 +1,9 @@ edge_tts==6.1.7 fairseq==0.12.2 faiss_cpu==1.7.4 -gradio==3.36.1 -gradio_client==0.2.9 +gradio==3.38.0 librosa==0.9.1 numpy==1.23.5 praat-parselmouth==0.4.3 -pyworld==0.3.3 +pyworld==0.3.4 torchcrepe==0.0.20