Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import torch
 import torchaudio
 import yaml
-from models.gense import N2S, S2S
+from models.gense_wavlm import N2S, S2S
 
 
 class AttrDict(dict):
@@ -30,8 +30,8 @@ def inference(noisy_path):
 
 from huggingface_hub import hf_hub_download
 config_path = hf_hub_download(repo_id="yaoxunji/gense", filename="gense.yaml")
-n2s_ckpt_path = hf_hub_download(repo_id="yaoxunji/gense", filename="n2s_xlsr.ckpt")
-s2s_ckpt_path = hf_hub_download(repo_id="yaoxunji/gense", filename="s2s_xlsr.ckpt")
+n2s_ckpt_path = hf_hub_download(repo_id="yaoxunji/gense", filename="n2s_wavlm.ckpt")
+s2s_ckpt_path = hf_hub_download(repo_id="yaoxunji/gense", filename="s2s_wavlm.ckpt")