Update README.md

README.md

@@ -5,7 +5,7 @@ base_model:
 - speechbrain/tts-hifigan-unit-hubert-l6-k100-ljspeech
 library_name: speechbrain
 ---
-# Basque Unit-HiFiGAN Vocoder (Voices: Mariana & Alex)
+# Basque Unit-HiFiGAN Vocoder (Voices: Maider & Antton)
 ## Model Summary
 
 This repository provides a Unit-HiFiGAN vocoder trained to synthesize high-fidelity Basque speech from discrete HuBERT-derived unit sequences. The model supports two speaker identities, Maider and Antton, using learned speaker-conditioning embeddings. It is compatible with HuBERT features extracted from layer 9 and clustered using a KMeans (k=1000) quantizer.
@@ -49,11 +49,11 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 SR = 16000
 
 # 1. Load HuBERT
-processor = Wav2Vec2Processor.from_pretrained("your-hubert-repo")
-hubert = HubertModel.from_pretrained("your-hubert-repo").to(DEVICE).eval()
+processor = Wav2Vec2Processor.from_pretrained("Ansu/HiFiGAN-Basque-Maider-Antton")
+hubert = HubertModel.from_pretrained("Ansu/HiFiGAN-Basque-Maider-Antton").to(DEVICE).eval()
 
 # 2. Load KMeans
-kmeans_path = hf_hub_download("your-hubert-repo", "kmeans/basque_hubert_k1000_L9.pt")
+kmeans_path = hf_hub_download("Ansu/HiFiGAN-Basque-Maider-Antton", "kmeans/basque_hubert_k1000_L9.pt")
 kmeans = joblib.load(kmeans_path)
 
 # 3. Load vocoder