Update InferenceInterfaces/ToucanTTSInterface.py

InferenceInterfaces/ToucanTTSInterface.py

@@ -24,15 +24,19 @@ class ToucanTTSInterface(torch.nn.Module):
 
     def __init__(self,
                  device="cpu",  # device that everything computes on. If a cuda device is available, this can speed things up by an order of magnitude.
-                 tts_model_path=os.path.join(MODELS_DIR, f"ToucanTTS_Meta", "best.pt"),  # path to the ToucanTTS checkpoint or just a shorthand if run standalone
-                 vocoder_model_path=os.path.join(MODELS_DIR, f"Vocoder", "best.pt"),  # path to the Vocoder checkpoint
+                 tts_model_path=None,  # path to the ToucanTTS checkpoint or just a shorthand if run standalone
+                 vocoder_model_path=None,  # path to the Vocoder checkpoint
                  language="eng",  # initial language of the model, can be changed later with the setter methods
                  ):
         super().__init__()
         self.device = device
-        if not tts_model_path.endswith(".pt"):
+        if tts_model_path is None:
+            tts_model_path = hf_hub_download(repo_id="Flux9665/ToucanTTS", filename="ToucanTTS.pt")
+        elif not tts_model_path.endswith(".pt"):
             # default to shorthand system
             tts_model_path = os.path.join(MODELS_DIR, f"ToucanTTS_{tts_model_path}", "best.pt")
+        if vocoder_model_path is None:
+            vocoder_model_path = hf_hub_download(repo_id="Flux9665/ToucanTTS", filename="Vocoder.pt")
 
         ################################
         #   build text to phone        #