Pre-download all the models

Pre-download all the models, using Repository from huggingface_hub

app.py

@@ -9,11 +9,14 @@ from typing import cast
 
 import gradio as gr
 from balacoon_tts import TTS
-from huggingface_hub import hf_hub_download, list_repo_files
+from huggingface_hub import Repository
 
 # global tts module, initialized from a model selected
 tts = None
-
+model_repo_dir = "models"
+model_repo = Repository(
+    local_dir=model_repo_dir, clone_from="balacoon/tts"
+)
 
 def main():
     logging.basicConfig(level=logging.INFO)
@@ -40,7 +43,7 @@ def main():
 
         with gr.Row():
             with gr.Column(variant="panel"):
-                repo_files = list_repo_files(repo_id="balacoon/tts")
+                repo_files = glob.glob(os.path.join(model_repo_dir, "*.addon"))
                 model_files = [x for x in repo_files if x.endswith("_cpu.addon")]
                 model_name = gr.Dropdown(
                     label="Model",
@@ -55,9 +58,7 @@ def main():
                 re-initializes tts object, gets list of
                 speakers that model supports and set them to `speaker`
                 """
-                model_path = hf_hub_download(
-                    repo_id="balacoon/tts", filename=model_name_str
-                )
+                model_path = os.path.join(model_repo_dir, model_name_str)
                 global tts
                 tts = TTS(model_path)
                 speakers = tts.get_speakers()