Update app.py

app.py

@@ -27,8 +27,10 @@ def download_model_and_config(gender):
         os.makedirs(dir_path)
     model_url = MALE_MODEL_URL if gender == "male" else FEMALE_MODEL_URL
     config_url = MALE_CONFIG_URL if gender == "male" else FEMALE_CONFIG_URL
+    onnx_model_url = MALE_ONNX_MODEL_URL if gender == "male" else FEMALE_ONNX_MODEL_URL
     download_url(model_url, dir_path, "model.pth")
     download_url(config_url, dir_path, "config.json")
+    download_url(onnx_model_url, dir_path, "model.onnx")
     return dir_path
 
 download_model_and_config("male")
@@ -46,7 +48,7 @@ def tts(text: str, voice: str = "Male", use_onnx: bool = True):
     text = text.lower()
 
     if use_onnx:
-        onnx_model_url = MALE_ONNX_MODEL_URL if voice == "Male" else FEMALE_ONNX_MODEL_URL
+        onnx_model_path =  f"{model_dir}/model.onnx"
 
         config = VitsConfig()
         config.load_json(config_file)
@@ -60,7 +62,7 @@ def tts(text: str, voice: str = "Male", use_onnx: bool = True):
         )
 
         vits = Vits.init_from_config(config)
-        vits.load_onnx(onnx_model_url)
+        vits.load_onnx(onnx_model_path)
 
         text_inputs = np.asarray(
             vits.tokenizer.text_to_ids(text),