Update app.py

app.py

@@ -13,22 +13,27 @@ MODEL_INFO = [
 # MODEL_NAMES = [info[0] for info in MODEL_INFO]
 
 MODEL_NAMES = [
-    "vits checkoint 57000",
+    "vits checkpoint 57000",
     # Add other model names similarly...
 ]
 
-
 MAX_TXT_LEN = 400
 TOKEN = os.getenv('HUGGING_FACE_HUB_TOKEN')
 
-# Download models
-for model_name, model_file, config_file, repo_name in MODEL_INFO:
-    os.makedirs(model_name, exist_ok=True)
-    print(f"|> Downloading: {model_name}")
+# # Download models
+# for model_name, model_file, config_file, repo_name in MODEL_INFO:
+#     os.makedirs(model_name, exist_ok=True)
+#     print(f"|> Downloading: {model_name}")
     
-    # Use hf_hub_download to download models from private Hugging Face repositories
-    hf_hub_download(repo_id=repo_name, filename=model_file, cache_dir=model_name, use_auth_token=TOKEN)
-    hf_hub_download(repo_id=repo_name, filename=config_file, cache_dir=model_name, use_auth_token=TOKEN)
+#     # Use hf_hub_download to download models from private Hugging Face repositories
+#     hf_hub_download(repo_id=repo_name, filename=model_file, use_auth_token=TOKEN)
+#     hf_hub_download(repo_id=repo_name, filename=config_file, use_auth_token=TOKEN)
+
+repo_name = "mhrahmani/persian-tts-vits-0"
+filename = "checkpoint_57000.pth"
+
+model_file = hf_hub_download(repo_name, filename, use_auth_token=TOKEN)
+config_file = hf_hub_download(repo_name, "config.json", use_auth_token=TOKEN)
 
 
 def synthesize(text: str, model_name: str) -> str:
@@ -37,7 +42,7 @@ def synthesize(text: str, model_name: str) -> str:
         text = text[:MAX_TXT_LEN]
         print(f"Input text was cut off as it exceeded the {MAX_TXT_LEN} character limit.")
     
-    synthesizer = Synthesizer(f"{model_name}/{model_file}", f"{model_name}/{config_file}")
+    synthesizer = Synthesizer(model_file, config_file)
     if synthesizer is None:
         raise NameError("Model not found")