deploy tts

app.py

@@ -3,10 +3,12 @@ from typing import Optional
 import gradio as gr
 import numpy as np
 from TTS.api import TTS
+from huggingface_hub import hf_hub_download
+
 
 MAX_TXT_LEN = 100
-# manager = ModelManager()
-model_name="DigitalUmuganda/Kinyarwanda_YourTTS"
+
+hf_hub_download(repo_id="DigitalUmuganda/Kinyarwanda_YourTTS")
 def generate_audio(text):
     if len(text) > MAX_TXT_LEN:
         text = text[:MAX_TXT_LEN]
@@ -22,8 +24,12 @@ def generate_audio(text):
     # )
     # if synthesizer is None:
     #     raise NameError("model not found")
-    tts = TTS(model_name)
-    wav = tts.tts(text, speaker=tts.speakers[0], language=tts.languages[0])
+    tts = TTS(model_path="Kinyarwanda_YourTTS/model.pth",
+            config_path="Kinyarwanda_YourTTS/config.json",
+            tts_speakers_file="Kinyarwanda_YourTTS/speakers.pth"
+            encoder_checkpoint="Kinyarwanda_YourTTS/SE_checkpoint.pth.tar",
+            encoder_config="Kinyarwanda_YourTTS/config_se.json",)
+    wav = tts.tts(text, speaker_wav="kinyarwanda_YourTTS/conditioning_audio.wav")
     return wav
     # with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp:
     #     synthesizer.save_wav(wav, fp)