download querier from hf directly; remove examples

app.py

@@ -140,18 +140,6 @@ with gr.Blocks() as demo:
         outputs=[out_audio, status]
     )
 
-    # 示例部分
-    gr.Examples(
-        examples=[
-            ["examples/noisy_1.wav", "audio", "examples/query_1.wav", ""],
-            ["examples/noisy_2.wav", "text", "", "提取人声"],
-            ["examples/noisy_3.wav", "text", "", "去除背景噪声"],
-        ],
-        inputs=[inp_audio, query_type, inp_query_audio, inp_query_text],
-        label="样例音频",
-        examples_per_page=3,
-    )
-
 # Queue: keep a small queue to avoid OOM
 demo.queue(max_size=8)  # 减少队列大小，因为现在需要更多资源
 demo.launch()

core/models/e2e/querier/passt.py

@@ -1,3 +1,4 @@
+from huggingface_hub import hf_hub_download
 import torch
 import torchaudio as ta
 from hear21passt.base import get_basic_model
@@ -17,7 +18,8 @@ class Passt(nn.Module):
         super().__init__()
 
         self.passt = laion_clap.CLAP_Module(enable_fusion=False, amodel='HTSAT-base')
-        self.passt.load_ckpt('checkpoints/querier/music_speech_epoch_15_esc_89.25.pt')
+        ckpt_path = hf_hub_download(repo_id="lukewys/laion_clap", filename="music_speech_epoch_15_esc_89.25.pt")
+        self.passt.load_ckpt(ckpt_path)
         self.resample = ta.transforms.Resample(
             orig_freq=original_fs, new_freq=passt_fs
         ).eval()