Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 import os
+import shutil
 from huggingface_hub import hf_hub_download
 from audio_separator.separator import Separator
 
 # --- Configuration ---
 REPO_ID = "anvuew/dereverb_room"
-# Filenames taken from your logs
 MODEL_FILENAME = "dereverb_room_anvuew_sdr_13.7432.ckpt" 
 CONFIG_FILENAME = "dereverb_room_anvuew.yaml" 
 # ---------------------
@@ -14,26 +14,43 @@ def inference(audio_path):
     if not audio_path:
         return None
 
-    print("Downloading model files if missing...")
-    model_path = hf_hub_download(repo_id=REPO_ID, filename=MODEL_FILENAME)
-    config_path = hf_hub_download(repo_id=REPO_ID, filename=CONFIG_FILENAME)
+    print("Downloading model files...")
+    # Download to a specific local directory to ensure they are together
+    os.makedirs("models", exist_ok=True)
+    model_path = hf_hub_download(repo_id=REPO_ID, filename=MODEL_FILENAME, local_dir="models")
+    config_path = hf_hub_download(repo_id=REPO_ID, filename=CONFIG_FILENAME, local_dir="models")
+
+    # Rename config to match model filename if they differ significantly, 
+    # though standard loading might just work if they are in the same dir.
+    # Let's try standard loading first by just pointing to the checkpoint.
 
     print("Initializing separator...")
-    # Initialize WITHOUT model paths first
     separator = Separator(
         output_dir=".",
         output_format="FLAC"
     )
 
-    print(f"Loading model: {MODEL_FILENAME}...")
-    # Load the specific model and config using their absolute downloaded paths
-    separator.load_model(model_filename=model_path, config_filename=config_path)
+    print(f"Loading model from {model_path}...")
+    # Simply pass the model path. The library usually looks for a .yaml 
+    # with a matching name or the specific config it needs in the same dir.
+    # If it strictly needs the config passed, we might need to use a different method
+    # but standard usage often just takes the model file if packaged correctly.
+    # Let's try to pass JUST the model path first as per common usage.
+    #
+    # NOTE: If this still fails because it can't find the config, we will need
+    # to rename 'dereverb_room_anvuew.yaml' to 'dereverb_room_anvuew_sdr_13.7432.yaml'
+    
+    expected_config_path = os.path.splitext(model_path)[0] + ".yaml"
+    if not os.path.exists(expected_config_path):
+         print(f"Renaming config to {expected_config_path} for auto-discovery...")
+         shutil.copy(config_path, expected_config_path)
+
+    separator.load_model(model_filename=model_path)
 
     print("Starting inference...")
     output_files = separator.separate(audio_path)
     
     print(f"Separation complete. Outputs: {output_files}")
-    # Return the first output file
     return output_files[0]
 
 with gr.Blocks(title="Dereverb Room Web UI") as demo: