Beehzod commited on
Commit
0272995
·
verified ·
1 Parent(s): f187ef9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/speecht5_finetuned_uz_customData2").to(device)
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[304]
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