kfahn commited on
Commit
3867b83
·
1 Parent(s): 42da1cb

Update app.py

Browse files

Try to add model fine-tuned for Czech language

Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -12,9 +12,11 @@ device = "cuda:0" if torch.cuda.is_available() else "cpu"
12
  asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base", device=device)
13
 
14
  # load text-to-speech checkpoint and speaker embeddings
15
- processor = SpeechT5Processor.from_pretrained("microsoft/speecht5_tts")
 
16
 
17
- model = SpeechT5ForTextToSpeech.from_pretrained("microsoft/speecht5_tts").to(device)
 
18
  vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan").to(device)
19
 
20
  embeddings_dataset = load_dataset("Matthijs/cmu-arctic-xvectors", split="validation")
 
12
  asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base", device=device)
13
 
14
  # load text-to-speech checkpoint and speaker embeddings
15
+ #processor = SpeechT5Processor.from_pretrained("microsoft/speecht5_tts")
16
+ processor = SpeechT5Processor.from_pretrained("kfahn/speecht5_finetuned_voxpopuli_cs")
17
 
18
+ model = SpeechT5ForTextToSpeech.from_pretrained("kfahn/speecht5_finetuned_voxpopuli_cs").to(device)
19
+ #model = SpeechT5ForTextToSpeech.from_pretrained("microsoft/speecht5_tts").to(device)
20
  vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan").to(device)
21
 
22
  embeddings_dataset = load_dataset("Matthijs/cmu-arctic-xvectors", split="validation")