Update app.py

app.py

@@ -7,12 +7,14 @@ from balacoon_tts import TTS
 
 # Global TTS module, initialized from a selected model
 tts = None
+speakers = []
+default_speaker = None
 
 def read_pdf(file):
     text = extract_text(file.name)
     return text
 
-def synthesize_audio(file, model_name_str, speaker_str):
+def synthesize_audio(file, speaker_str):
     text_str = read_pdf(file)
     if len(text_str) > 1024:
         text_str = text_str[:1024]
@@ -28,19 +30,17 @@ def main():
 
     def set_model(model_name_str: str):
         model_path = hf_hub_download(repo_id="balacoon/tts", filename=model_name_str)
-        global tts
+        global tts, speakers, default_speaker
         tts = TTS(model_path)
         speakers = tts.get_speakers()
         default_speaker = speakers[-1]
-        speaker_dropdown.choices = speakers
-        return speakers, default_speaker
 
-    speaker_dropdown = gr.inputs.Dropdown(label="Speaker", choices=[])
+    speaker_dropdown = gr.inputs.Dropdown(label="Speaker", choices=speakers)
 
     audio = gr.outputs.Audio(label="Generated Audio", type="numpy")
 
-    def generate_audio(file, model_name_str, speaker_str):
-        return synthesize_audio(file, model_name_str, speaker_str)
+    def generate_audio(file, speaker_str):
+        return synthesize_audio(file, speaker_str)
 
     iface = gr.Interface(
         fn=generate_audio,
@@ -51,7 +51,7 @@ def main():
         debug=True
     )
 
-    model_name_dropdown.observe(set_model, "value")
+    model_name_dropdown.set_action(set_model)
 
     iface.launch()