Update app.py

app.py

@@ -19,7 +19,8 @@ BASE_PATH_MODEL = os.path.join(BASE_PATH, "Model")
 URL_PIPER_DOWNLOAD = "https://github.com/rhasspy/piper/releases/download/v1.2.0/piper_amd64.tar.gz"
 
 # TTS model files
-URL_TTS_ONNX = "https://huggingface.co/aslanovaf/tts_vits_azerbaijani/resolve/main/last.onnx"
+URL_TTS_ONNX = "https://huggingface.co/CavidanZ/TTS-azerbaijani-model/resolve/main/last.onnx" 
+URL_TTS_ONNX_2 = "https://huggingface.co/CavidanZ/TTS-azerbaijani-model/resolve/main/last2.onnx" 
 TMP_PIPER_FILENAME = os.path.join(BASE_PATH, "piper.tgz")
 
 headers = {"Authorization": f"Bearer {HF_TOKEN}"}
@@ -67,6 +68,21 @@ if os.path.exists(BASE_PATH_MODEL) == False:
     else:
         st.markdown(f"Failed to download TTS json from {URL_TTS_ONNX}.json (Status code: {response.status_code})")
 
+    response = requests.get(URL_TTS_ONNX_2, headers=headers) 
+    if response.status_code == 200:
+        with open(os.path.join(BASE_PATH_MODEL, "last2.onnx"), 'wb') as f:
+            f.write(response.content)
+    else:
+        st.markdown(f"Failed to download TTS from {URL_TTS_ONNX} (Status code: {response.status_code})")
+
+
+    response = requests.get((URL_TTS_ONNX_2 + ".json"), headers=headers) 
+    if response.status_code == 200:
+        with open(os.path.join(BASE_PATH_MODEL, "last2.onnx.json"), 'wb') as f:
+            f.write(response.content)
+    else:
+        st.markdown(f"Failed to download TTS json from {URL_TTS_ONNX}.json (Status code: {response.status_code})")
+
 
 ###########################
 # MODEL DOWNLOAD FINISHED #
@@ -93,7 +109,8 @@ st.header('A text-to-speech (TTS) model in Azerbaijani language')
 with st.form("my_form"):
    option = st.selectbox(
      'Choose the model.',
-     ['Model 1'])
+     ['Model 1'],
+     ['Model 2'])
 
    text = st.text_area("Text to generate audio from:",max_chars=500)
    submitted = st.form_submit_button("Submit")
@@ -105,11 +122,12 @@ with st.form("my_form"):
         # Set Piper TTS command based on choice
         PIPER_CMD = os.path.join(BASE_PATH,"piper","piper")
         SPEAKER_ID = "0"
-        MODEL = "last.onnx"
 
         match option:
             case "Model 1":
                 MODEL = "last.onnx"
+            case "Model 2":
+                MODEL = "last2.onnx"
 
         cmd = "echo '" + text + "' | " + BASE_PATH + "/piper/piper --model " + os.path.join(BASE_PATH_MODEL, MODEL) + " --speaker " + SPEAKER_ID + " --output_file " + filename.name