Update app.py

app.py

@@ -17,26 +17,18 @@ MODEL_NAMES = [info[0] for info in MODEL_INFO]
 MAX_TXT_LEN = 400
 TOKEN = os.environ.get('HUGGING_FACE_HUB_TOKEN')  # Replace with the environment variable containing your token, if different
 
-# 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 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)
 
 # Verify if the files are downloaded correctly
 for model_name, model_file, config_file, repo_name in MODEL_INFO:
-    os.makedirs(model_name, exist_ok=True)
+    # os.makedirs(model_name, exist_ok=True)
     print(f"|> Downloading: {model_name}")
     
-    model_file_path = hf_hub_download(repo_id=repo_name, filename=model_file, cache_dir=model_name, use_auth_token=TOKEN)
-    config_file_path = hf_hub_download(repo_id=repo_name, filename=config_file, cache_dir=model_name, use_auth_token=TOKEN)
+    model_file_path = hf_hub_download(repo_id=repo_name, filename=model_file, use_auth_token=TOKEN)
+    config_file_path = hf_hub_download(repo_id=repo_name, filename=config_file, use_auth_token=TOKEN)
     
-    # Check if the files exist after download
-    if not os.path.exists(model_file_path) or not os.path.exists(config_file_path):
-        raise FileNotFoundError(f"Failed to download files for {model_name}. Please check the repository and file names.")
+    # # Check if the files exist after download
+    # if not os.path.exists(model_file_path) or not os.path.exists(config_file_path):
+    #     raise FileNotFoundError(f"Failed to download files for {model_name}. Please check the repository and file names.")
 
 
 def synthesize(text: str, model_name: str) -> str:
@@ -48,7 +40,7 @@ def synthesize(text: str, model_name: str) -> str:
     # Extract model_file and config_file based on the model_name
     model_file, config_file = next((model_file, config_file) for name, model_file, config_file, _ in MODEL_INFO if name == model_name)
     
-    synthesizer = Synthesizer(f"{model_name}/{model_file}", f"{model_name}/{config_file}")
+    synthesizer = Synthesizer(model_file_path, config_file_path)
     if synthesizer is None:
         raise NameError("Model not found")