fix examples

app.py

@@ -100,12 +100,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="vit_l_16_cqt")
     examples = []
     example_audios = find_audio_files()
-    model_num = len(models)
     for audio in example_audios:
-        examples.append([audio, models[random.randint(0, model_num - 1)]])
+        examples.append([audio, models[0]])
 
     with gr.Blocks() as demo:
         gr.Interface(

utils.py

@@ -29,7 +29,7 @@ def find_audio_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:
@@ -40,11 +40,15 @@ def get_modelist(model_dir=MODEL_DIR):
     for entry in entries:
         full_path = os.path.join(model_dir, entry)
         if entry == ".git" or entry == "examples":
-            print(f"Skip .git or examples dir: {full_path}")
+            print(f"Skip .git / examples dir: {full_path}")
             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