Update app.py

app.py

@@ -209,17 +209,12 @@ load_resources()
 model_files = glob.glob(os.path.join(MODELS_DIR, "*.pt"))
 model_choices = [os.path.basename(f) for f in model_files]
 
-# Ensure 'musicflow_b.pt' is the default choice if it exists
-default_model = 'musicflow_b.pt'
-if default_model in model_choices:
-    model_choices.remove(default_model)
-    model_choices.insert(0, default_model)
+print(f"Found model files: {model_choices}")  # Debug print
 
 # Handle the case where no models are found
 if not model_choices:
     print("No model files found in the specified directory.")
     model_choices = ["No models available"]
-    default_model = "No models available"
 
 # Set up dark grey theme
 theme = gr.themes.Monochrome(
@@ -240,7 +235,7 @@ with gr.Blocks(theme=theme) as iface:
         """)
     
     with gr.Row():
-        model_dropdown = gr.Dropdown(choices=model_choices, label="Select Model", value=default_model)
+        model_dropdown = gr.Dropdown(choices=model_choices, label="Select Model", value=model_choices[0])
     
     with gr.Row():
         prompt = gr.Textbox(label="Prompt")
@@ -264,13 +259,11 @@ with gr.Blocks(theme=theme) as iface:
     model_dropdown.change(on_model_change, inputs=[model_dropdown])
     generate_button.click(generate_music, inputs=[prompt, seed, cfg_scale, steps, duration], outputs=[output_status, output_audio])
 
-    # Load default model on startup only if it exists
-    if default_model != "No models available":
-        default_model_path = os.path.join(MODELS_DIR, default_model)
-        if os.path.exists(default_model_path):
-            iface.load(lambda: load_model(default_model), inputs=None, outputs=None)
+    # Load first available model on startup
+    if model_choices[0] != "No models available":
+        iface.load(lambda: load_model(model_choices[0]), inputs=None, outputs=None)
     else:
-        print("No default model available to load.")
+        print("No models available to load.")
 
 # Launch the interface
-iface.launch()
+iface.launch()