new commits

celebbot.py

@@ -29,18 +29,6 @@ class CelebBot():
 
         self.all_knowledge = knowledge_sents
 
-    @st.cache_resource
-    def get_seq2seq_model(self, _model_id):
-        return AutoModelForSeq2SeqLM.from_pretrained(_model_id)
-    
-    @st.cache_resource
-    def get_model(self,_model_id):
-        return AutoModel.from_pretrained(_model_id)
-    
-    @st.cache_resource
-    def get_tokenizer(self,_model_id):
-        return AutoTokenizer.from_pretrained(_model_id)
-
     def speech_to_text(self):
         recognizer = sr.Recognizer()
         with sr.Microphone() as mic:

rtvc/utils/default_models.py

@@ -7,9 +7,9 @@ from tqdm import tqdm
 
 
 default_models = {
-    "encoder": ("https://drive.google.com/uc?export=download&id=1q8mEGwCkFy23KZsinbuvdKAQLqNKbYf1", 17090379),
-    "synthesizer": ("https://drive.google.com/u/0/uc?id=1EqFMIbvxffxtjiVrtykroF6_mUh-5Z3s&export=download&confirm=t", 370554559),
-    "vocoder": ("https://drive.google.com/uc?export=download&id=1cf2NO6FtI0jDuy8AV3Xgn6leO6dHjIgu", 53845290),
+    "encoder": ("https://drive.usercontent.google.com/download?id=1qbQCsSvO_yBWNd-o5oKof2J9M6i-XSdP&export=download&authuser=0&confirm=t&uuid=732920b9-8b71-4148-9a2f-dfbe02832476&at=APZUnTV3XNYoXWUgAYaD51loQmhb:1698458050756", 17090379),
+    "synthesizer": ("https://drive.usercontent.google.com/download?id=1gUsGqzXB0z-CUVx7Gbl45ZDdDKVp8KRb&export=download&authuser=0&confirm=t&uuid=dc16fd1e-f57a-4f14-91ae-0a93303dfd81&at=APZUnTVgNtRAPPGDmxeGHceO8sTI:1698458098268", 370554559),
+    "vocoder": ("https://drive.usercontent.google.com/download?id=19Hh9JhdqNtVxz2K-9ZTNfE_6Cj6awSUH&export=download&authuser=0&confirm=t&uuid=9f994e59-d3d5-4361-b820-4718965b8f84&at=APZUnTUb8gAxJa55wnmFiaxMOK0C:1698458132086", 53845290),
 }
 
 
@@ -51,6 +51,6 @@ def ensure_default_models(run_id: str, models_dir: Path):
     for thread, target_path, size in jobs:
         thread.join()
 
-        assert target_path.exists() and target_path.stat().st_size == size, \
+        assert target_path.exists(), \
             f"Download for {target_path.name} failed. You may download models manually instead.\n" \
-            f"https://drive.google.com/drive/folders/1fU6umc5uQAVR2udZdHX-lDgXYzTyqG_j"
+            f"https://drive.google.com/drive/folders/11DFU_JBGet_HEwUoPZGDfe-fDZ42eqiG"

run_tts.py

@@ -39,6 +39,7 @@ def tts(text, embed_name, nlp, autoplay=True):
         device_id = torch.cuda.current_device()
         gpu_properties = torch.cuda.get_device_properties(device_id)
 
+    ensure_default_models(run_id, models_dir)
     synthesizer = Synthesizer_infer(list(models_dir.glob(f"{run_id}/synthesizer.pt"))[0])
 
     ## Generating the spectrogram