Update components/semantic_extractor/ssl_model.py

components/semantic_extractor/ssl_model.py

@@ -47,6 +47,7 @@ def get_ssl_model(ckpt_path, km_path, device='cuda', type='xlsr'):
         model = model.requires_grad_(False)
     else:
         raise NotImplementedError
+    km_path = hf_hub_download(repo_id="yaoxunji/gense", filename="wavlm_km.mdl")
     km_model = ApplyKmeans(km_path, device)
     return model, km_model