tae98 commited on
Commit
34b8876
·
1 Parent(s): df975d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,6 @@ processor = SpeechT5Processor.from_pretrained("microsoft/speecht5_tts")
16
 
17
  model = SpeechT5ForTextToSpeech.from_pretrained("sanchit-gandhi/speecht5_tts_vox_nl").to(device)
18
  vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan").to(device)
19
- translator= SpeechT5HifiGan.from_pretrained("sanchit-gandhi/speecht5_tts_vox_nl").to(device)
20
 
21
  embeddings_dataset = load_dataset("Matthijs/cmu-arctic-xvectors", split="validation")
22
  speaker_embeddings = torch.tensor(embeddings_dataset[7306]["xvector"]).unsqueeze(0)
@@ -24,6 +23,7 @@ speaker_embeddings = torch.tensor(embeddings_dataset[7306]["xvector"]).unsqueeze
24
 
25
  def translate(audio):
26
  outputs = asr_pipe(audio, max_new_tokens=256, generate_kwargs={"task": "translate"})
 
27
  return outputs["text"]
28
 
29
 
 
16
 
17
  model = SpeechT5ForTextToSpeech.from_pretrained("sanchit-gandhi/speecht5_tts_vox_nl").to(device)
18
  vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan").to(device)
 
19
 
20
  embeddings_dataset = load_dataset("Matthijs/cmu-arctic-xvectors", split="validation")
21
  speaker_embeddings = torch.tensor(embeddings_dataset[7306]["xvector"]).unsqueeze(0)
 
23
 
24
  def translate(audio):
25
  outputs = asr_pipe(audio, max_new_tokens=256, generate_kwargs={"task": "translate"})
26
+ print(outputs["text"])
27
  return outputs["text"]
28
 
29