diff --git a/tools/uvr5/webui.py b/tools/uvr5/webui.py index df23522e..7749fbed 100644 --- a/tools/uvr5/webui.py +++ b/tools/uvr5/webui.py @@ -34,10 +34,10 @@ def uvr(model_name, inp_root, save_root_vocal, paths, save_root_ins, agg, format is_hp3 = "HP3" in model_name if model_name == "onnx_dereverb_By_FoxJoy": pre_fun = MDXNetDereverb(15) - elif model_name == "Bs_Roformer": + elif model_name == "Bs_Roformer" or "bs_roformer" in model_name.lower(): func = BsRoformer_Loader pre_fun = func( - model_path = os.path.join(weight_uvr5_root, "Bs_Roformer.pth"), + model_path = os.path.join(weight_uvr5_root, model_name + ".pth"), device = device, ) else: