xiankai123 commited on
Commit
de20f0f
1 Parent(s): f8f5e1d
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,8 +12,8 @@ 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 MMS VITS model
15
- model = VitsModel.from_pretrained("mms-1b-fl102")
16
- tokenizer = VitsTokenizer.from_pretrained("mms-1b-fl102")
17
 
18
  embeddings_dataset = load_dataset("Matthijs/cmu-arctic-xvectors", split="validation")
19
  speaker_embeddings = torch.tensor(embeddings_dataset[7306]["xvector"]).unsqueeze(0)
 
12
  asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base", device=device)
13
 
14
  # load MMS VITS model
15
+ model = VitsModel.from_pretrained("facebook/mms-1b-fl102")
16
+ tokenizer = VitsTokenizer.from_pretrained("facebook/mms-1b-fl102")
17
 
18
  embeddings_dataset = load_dataset("Matthijs/cmu-arctic-xvectors", split="validation")
19
  speaker_embeddings = torch.tensor(embeddings_dataset[7306]["xvector"]).unsqueeze(0)