Update app.py

app.py

@@ -3,7 +3,6 @@ import torch
 import shutil
 import logging
 import gradio as gr
-import pandas as pd
 
 from PolUVR.separator import Separator
 
@@ -14,21 +13,27 @@ use_autocast = device == "cuda"
 #     Roformer Models     #
 #=========================#
 ROFORMER_MODELS = {
-    'BS Roformer | sdr 10.5309 - by Viperx': 'model_bs_roformer_ep_937_sdr_10.5309.ckpt',
-    'BS Roformer | sdr 12.9628 - by Viperx': 'model_bs_roformer_ep_368_sdr_12.9628.ckpt',
-    'BS Roformer | sdr 12.9755 - by Viperx': 'model_bs_roformer_ep_317_sdr_12.9755.ckpt',
-    'BS Roformer | De-Reverb - by anvuew': 'deverb_bs_roformer_8_384dim_10depth.ckpt',
-    'Mel Roformer | sdr 11.4360 - by Viperx': 'model_mel_band_roformer_ep_3005_sdr_11.4360.ckpt',
-    'Mel Roformer | Vocals - by KimberleyJensen': 'vocals_mel_band_roformer.ckpt',
-    'Mel Roformer | Crowd - by Aufr33 and Viperx': 'mel_band_roformer_crowd_aufr33_viperx_sdr_8.7144.ckpt',
-    'Mel Roformer | Karaoke - by Aufr33 and Viperx': 'mel_band_roformer_karaoke_aufr33_viperx_sdr_10.1956.ckpt',
-    'Mel Roformer | Denoise - by Aufr33': 'denoise_mel_band_roformer_aufr33_sdr_27.9959.ckpt',
-    'Mel Roformer | Denoise Aggr - by Aufr33': 'denoise_mel_band_roformer_aufr33_aggr_sdr_27.9768.ckpt',
-    'MelBand Roformer Kim | Inst V1 - by Unwa': 'melband_roformer_inst_v1.ckpt',
-    'MelBand Roformer Kim | Inst V1 (E) - by Unwa': 'melband_roformer_inst_v1e.ckpt',
-    'MelBand Roformer Kim | Inst V2 - by Unwa': 'melband_roformer_inst_v2.ckpt',
-    'MelBand Roformer Kim | InstVoc Duality V1 - by Unwa': 'melband_roformer_instvoc_duality_v1.ckpt',
-    'MelBand Roformer Kim | InstVoc Duality V2 - by Unwa': 'melband_roformer_instvox_duality_v2.ckpt',
+    'BS Roformer | sdr 10.5309 — by Viperx': 'model_bs_roformer_ep_937_sdr_10.5309.ckpt',
+    'BS Roformer | sdr 12.9628 — by Viperx': 'model_bs_roformer_ep_368_sdr_12.9628.ckpt',
+    'BS Roformer | sdr 12.9755 — by Viperx': 'model_bs_roformer_ep_317_sdr_12.9755.ckpt',
+    'BS Roformer | Chorus — by Sucial': 'model_chorus_bs_roformer_ep_267_sdr_24.1275.ckpt',
+    'BS Roformer | DeReverb — by anvuew': 'deverb_bs_roformer_8_384dim_10depth.ckpt',
+    'Mel Roformer | sdr 11.4360 — by Viperx': 'model_mel_band_roformer_ep_3005_sdr_11.4360.ckpt',
+    'Mel Roformer | Vocals — by KimberleyJensen': 'vocals_mel_band_roformer.ckpt',
+    'Mel Roformer | Crowd — by Aufr33 and Viperx': 'mel_band_roformer_crowd_aufr33_viperx_sdr_8.7144.ckpt',
+    'Mel Roformer | Karaoke — by Aufr33 and Viperx': 'mel_band_roformer_karaoke_aufr33_viperx_sdr_10.1956.ckpt',
+    'Mel Roformer | DeNoise — by Aufr33': 'denoise_mel_band_roformer_aufr33_sdr_27.9959.ckpt',
+    'Mel Roformer | DeNoise Aggr — by Aufr33': 'denoise_mel_band_roformer_aufr33_aggr_sdr_27.9768.ckpt',
+    'MelBand Roformer | Aspiration — by Sucial': 'aspiration_mel_band_roformer_sdr_18.9845.ckpt',
+    'MelBand Roformer | Aspiration Aggr — by Sucial': 'aspiration_mel_band_roformer_less_aggr_sdr_18.1201.ckpt',
+    'MelBand Roformer | DeReverb — by Anvuew': 'dereverb_mel_band_roformer_anvuew_sdr_19.1729.ckpt',
+    'MelBand Roformer | DeReverb Aggr — by Anvuew': 'dereverb_mel_band_roformer_less_aggressive_anvuew_sdr_18.8050.ckpt',
+    'MelBand Roformer | DeReverb-DeEcho v1 — by Sucial': 'dereverb-echo_mel_band_roformer_sdr_10.0169.ckpt',
+    'MelBand Roformer | DeReverb-DeEcho v2 — by Sucial': 'dereverb_echo_mbr_v2_sdr_dry_13.4843.ckpt',
+    'MelBand Roformer Kim | Inst V1 — by Unwa': 'melband_roformer_inst_v1.ckpt',
+    'MelBand Roformer Kim | Inst V1 (E) — by Unwa': 'melband_roformer_inst_v1e.ckpt',
+    'MelBand Roformer Kim | Inst V2 — by Unwa': 'melband_roformer_inst_v2.ckpt',
+    'MelBand Roformer Kim | InstVoc Duality — by Unwa': 'melband_roformer_instvoc_duality_v1.ckpt',
 }
 #=========================#
 #      MDX23C Models      #
@@ -123,42 +128,6 @@ DEMUCS_MODELS = [
     'htdemucs_ft.yaml',
 ]
 
-def load_data(file_path):
-    df = pd.read_csv(file_path)
-    return df
-
-def sort_data(architecture, purpose, sdr_metric):
-    df = load_data("models.csv")
-
-    if architecture != "All":
-        df = df[df['Architecture'] == architecture]
-    if purpose != "All":
-        df = df[df['Purpose'] == purpose]
-    
-    return df
-
-def create_leaderboard_tab():
-    architectures = ["All", "Roformer", "MDXC", "MDX", "VR-ARCH", "DEMUCS", "Other"]
-    purposes = ["All", "VocInst", "DeEcho", "DeReverb", "DeNoise", "Karaoke", "Crowd", "Other"]
-    
-    with gr.Row():
-        architecture_dropdown = gr.Dropdown(choices=architectures, label="Architecture")
-        purpose_dropdown = gr.Dropdown(choices=purposes, label="Purpose")
-    
-    with gr.Row():
-        sort_button = gr.Button("Sort Models", variant="primary")
-    
-    with gr.Row():
-        leaderboard_output = gr.Dataframe(label="Leaderboard", interactive=False)
-    
-    sort_button.click(
-        sort_data,
-        inputs=[architecture_dropdown, purpose_dropdown],
-        outputs=leaderboard_output
-    )
-
-    return leaderboard_output
-
 def print_message(input_file, model_name):
     """Prints information about the audio separation process."""
     base_name = os.path.splitext(os.path.basename(input_file))[0]
@@ -522,11 +491,6 @@ with gr.Blocks(
             with gr.Row():
                 guitar_stem = gr.Textbox(value="NAME_(STEM)_MODEL", label="Guitar Stem", info="Output example: Music_(Guitar)_BS-Roformer-Viperx-1297", placeholder="NAME_(STEM)_MODEL")
                 piano_stem = gr.Textbox(value="NAME_(STEM)_MODEL", label="Piano Stem", info="Output example: Music_(Piano)_BS-Roformer-Viperx-1297", placeholder="NAME_(STEM)_MODEL")
-
-    with gr.Tab("Leaderboard"):
-        gr.Markdown("## Model Leaderboard")
-        gr.Markdown("Sort models by architecture, purpose, and SDR metrics.")
-        create_leaderboard_tab()
     
     with gr.Tab("Credits"):
         gr.Markdown(