fix examples

app.py

@@ -115,12 +115,11 @@ def infer(wav_path: str, log_name: str, folder_path=TEMP_DIR):
 
 if __name__ == "__main__":
     warnings.filterwarnings("ignore")
-    models = get_modelist()
+    models = get_modelist(assign_model="Swin_T_mel")
     examples = []
     example_wavs = find_wav_files()
-    model_num = len(models)
     for wav in example_wavs:
-        examples.append([wav, models[random.randint(0, model_num - 1)]])
+        examples.append([wav, models[0]])
 
     with gr.Blocks() as demo:
         gr.Interface(

utils.py

@@ -29,7 +29,7 @@ def find_wav_files(folder_path=f"{MODEL_DIR}/examples"):
     return wav_files
 
 
-def get_modelist(model_dir=MODEL_DIR):
+def get_modelist(model_dir=MODEL_DIR, assign_model=""):
     try:
         entries = os.listdir(model_dir)
     except OSError as e:
@@ -44,7 +44,11 @@ def get_modelist(model_dir=MODEL_DIR):
             continue
 
         if os.path.isdir(full_path):
-            output.append(os.path.basename(full_path))
+            model = os.path.basename(full_path)
+            if assign_model and assign_model.lower() in model:
+                output.insert(0, model)
+            else:
+                output.append(model)
 
     return output