Update app.py

app.py

@@ -21,6 +21,7 @@ URL_PIPER_DOWNLOAD = "https://github.com/rhasspy/piper/releases/download/v1.2.0/
 # TTS model files
 URL_TTS_ONNX = "https://huggingface.co/spaces/CavidanZ/TTS-azerbaijani-model/resolve/main/last.onnx" 
 URL_TTS_ONNX_2 = "https://huggingface.co/spaces/CavidanZ/TTS-azerbaijani-model/resolve/main/last2.onnx" 
+URL_TTS_FH = "https://huggingface.co/spaces/CavidanZ/TTS-azerbaijani-model/resolve/main/last_FH.onnx"
 URL_TACOTRON2 = "https://huggingface.co/spaces/CavidanZ/TTS-azerbaijani-model/resolve/main/Audiobook_based"
 TMP_PIPER_FILENAME = os.path.join(BASE_PATH, "piper.tgz")
 
@@ -85,6 +86,23 @@ 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})")
 
+    # --- Download TTS model - VITS Version FH --- #
+    response = requests.get(URL_TTS_FH, 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_FH + ".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 #
 ###########################
@@ -105,7 +123,7 @@ st.header('A text-to-speech (TTS) model in Azerbaijani language')
 with st.form("my_form"):
    option = st.selectbox(
      'Choose the model.',
-     ('VITS Model 1','VITS Model 2'))
+     ('VITS Model 1','VITS Model 2','FH'))
 
    text = st.text_area("Text to generate audio from:",max_chars=500)
    # Default parameter values
@@ -124,6 +142,9 @@ with st.form("my_form"):
                 MODEL = "last2.onnx"
             case "VITS Model 2":
                 MODEL = "last.onnx"
+            case "FH":
+                MODEL = "last_FH.onnx"
+                
 
         cmd = "echo '" + text + "' | " + BASE_PATH + "/piper/piper --model " + os.path.join(BASE_PATH_MODEL, MODEL) + " --speaker " + SPEAKER_ID + " --output_file " + filename.name + " --length-scale " + str(length_scale)