feat: add merge model

app.py

@@ -14,7 +14,11 @@ OmegaConf.register_new_resolver("load_vosk", load_vosk)
 models_config = OmegaConf.to_object(OmegaConf.load("configs/models.yaml"))
 
 def automatic_speech_recognition(model_id: str, dialect_id: str, audio_data: str):
-    model = models_config[model_id]["model"][dialect_id]
+    if isinstance(models_config[model_id]["model"], dict):
+        model = models_config[model_id]["model"][dialect_id]
+    else: 
+        model = models_config[model_id]["model"]
+        
     sample_rate, audio_array = audio_data
     if audio_array.ndim == 2:
         audio_array = audio_array[:, 0]
@@ -51,6 +55,9 @@ def automatic_speech_recognition(model_id: str, dialect_id: str, audio_data: str
 def when_model_selected(model_id: str):
     model_config = models_config[model_id]
 
+    if "dialect_mapping" not in model_config["dialect_mapping"]:
+        return gr.update(visible=False)
+
     dialect_drop_down_choices = [
         (k, v) for k, v in model_config["dialect_mapping"].items()
     ]
@@ -58,6 +65,7 @@ def when_model_selected(model_id: str):
     return gr.update(
         choices=dialect_drop_down_choices,
         value=dialect_drop_down_choices[0][1],
+        visible=True,
     )
 
 
@@ -103,8 +111,8 @@ with demo:
         # 臺灣客語語音辨識系統
         ### Taiwanese Hakka Automatic-Speech-Recognition System
         ### 研發
-        - **[李鴻欣 Hung-Shin Lee](mailto:hungshinlee@gmail.com)（聯和科創 United Link Co., Ltd.）**
-        - **[陳力瑋 Li-Wei Chen](mailto:wayne900619@gmail.com)（聯和科創 United Link Co., Ltd.）**
+        - **[李鴻欣 Hung-Shin Lee](mailto:hungshinlee@gmail.com)（[聯和科創](https://www.104.com.tw/company/1a2x6bmu75)）**
+        - **[陳力瑋 Li-Wei Chen](mailto:wayne900619@gmail.com)（[聯和科創](https://www.104.com.tw/company/1a2x6bmu75)）**
         """
     )
     gr.Interface(

configs/models.yaml

@@ -12,4 +12,7 @@ htia-0.1:
     大埔: htia_dapu
     饒平: htia_raoping
     詔安: htia_zhaoan
-    南四縣: htia_nansixian
+    南四縣: htia_nansixian
+
+htia-0.1m:
+  model: ${load_vosk:formospeech/kaldi-taiwanese-hakka-merge}