nambn0321 commited on
Commit
c293d71
·
verified ·
1 Parent(s): ff98b86

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ from transformers import SpeechT5Processor, SpeechT5ForTextToSpeech
5
  from transformers.models.speecht5 import SpeechT5HifiGan
6
 
7
  # Load model and processor
8
- processor = SpeechT5Processor.from_pretrained("nambn0321/TTS_british_2nd_attempt")
9
- model = SpeechT5ForTextToSpeech.from_pretrained("nambn0321/TTS_british_2nd_attempt", use_safetensors=True, trust_remote_code=True)
10
  vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan")
11
 
12
  # Move to CUDA if available
 
5
  from transformers.models.speecht5 import SpeechT5HifiGan
6
 
7
  # Load model and processor
8
+ processor = SpeechT5Processor.from_pretrained("nambn0321/T5_US_accent_3")
9
+ model = SpeechT5ForTextToSpeech.from_pretrained("nambn0321/T5_US_accent_3", use_safetensors=True, trust_remote_code=True)
10
  vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan")
11
 
12
  # Move to CUDA if available