Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 14 |
def load_models_and_data():
|
| 15 |
model_name = "microsoft/speecht5_tts"
|
| 16 |
processor = SpeechT5Processor.from_pretrained(model_name)
|
| 17 |
-
model = SpeechT5ForTextToSpeech.from_pretrained("Beehzod/
|
| 18 |
vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan").to(device)
|
| 19 |
|
| 20 |
spk_model_name = "speechbrain/spkrec-xvect-voxceleb"
|
|
@@ -26,7 +26,7 @@ def load_models_and_data():
|
|
| 26 |
|
| 27 |
# Load a sample from a dataset for default embedding
|
| 28 |
dataset = load_dataset("Beehzod/UzTTS_data", split="train")
|
| 29 |
-
example = dataset[
|
| 30 |
|
| 31 |
return model, processor, vocoder, speaker_model, example
|
| 32 |
|
|
|
|
| 14 |
def load_models_and_data():
|
| 15 |
model_name = "microsoft/speecht5_tts"
|
| 16 |
processor = SpeechT5Processor.from_pretrained(model_name)
|
| 17 |
+
model = SpeechT5ForTextToSpeech.from_pretrained("Beehzod/uz_2301_3_tts").to(device)
|
| 18 |
vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan").to(device)
|
| 19 |
|
| 20 |
spk_model_name = "speechbrain/spkrec-xvect-voxceleb"
|
|
|
|
| 26 |
|
| 27 |
# Load a sample from a dataset for default embedding
|
| 28 |
dataset = load_dataset("Beehzod/UzTTS_data", split="train")
|
| 29 |
+
example = dataset[30]
|
| 30 |
|
| 31 |
return model, processor, vocoder, speaker_model, example
|
| 32 |
|