fix examples

app.py

@@ -166,19 +166,18 @@ if __name__ == "__main__":
 
         os.environ["PATH"] = f"{folder_path}:{os.environ.get('PATH', '')}"
 
-    models = get_modelist()
+    models = get_modelist(assign_model="VGG19_BN_cqt")
     examples = []
     example_mp3s = find_mp3_files()
-    model_num = len(models)
     for mp3 in example_mp3s:
-        examples.append([mp3, models[random.randint(0, model_num - 1)]])
+        examples.append([mp3, models[0]])
 
     with gr.Blocks() as demo:
         gr.Interface(
             fn=inference,
             inputs=[
                 gr.Audio(label="Upload MP3", type="filepath"),
-                gr.Dropdown(choices=models, label="Select a model", value=models[6]),
+                gr.Dropdown(choices=models, label="Select a model", value=models[0]),
             ],
             outputs=[
                 gr.Textbox(label="Audio filename", show_copy_button=True),

utils.py

@@ -28,7 +28,7 @@ def find_mp3_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:
@@ -43,7 +43,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