Update app.py

app.py

@@ -73,10 +73,22 @@ def gen_text(audio_filepath, action, source_lang, target_lang):
 def gen_speech(text, lang):
     set_seed(555)  # Make it deterministic
 
-    if lang=="en": 
-        model = "facebook/mms-tts-eng"
-    elif lang=="fr":
-        model = "facebook/mms-tts-fra"
+    match lang:
+        case "en":
+             model = "facebook/mms-tts-eng"
+        case "fr":
+             model = "facebook/mms-tts-fra"
+        case "de":
+             model = "facebook/mms-tts-deu"
+        case "es":
+             model = "facebook/mms-tts-spa"
+        case _:
+            model = "facebook/mms-tts-eng"
+    
+    # if lang=="en": 
+    #     model = "facebook/mms-tts-eng"
+    # elif lang=="fr":
+    #     model = "facebook/mms-tts-fra"
 
     # load TTS model
     tts_model = VitsModel.from_pretrained(model)