divakaivan commited on
Commit
5f151ed
·
verified ·
1 Parent(s): 1396b37

revert back.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def predict(text, speaker):
49
  else:
50
  speaker_embedding = np.load(speaker_embeddings[speaker[:3]])
51
 
52
- speaker_embedding = torch.tensor([1, 512]).unsqueeze(0)
53
 
54
  speech = model.generate_speech(input_ids, speaker_embedding, vocoder=vocoder)
55
 
 
49
  else:
50
  speaker_embedding = np.load(speaker_embeddings[speaker[:3]])
51
 
52
+ speaker_embedding = torch.tensor(speaker_embedding).unsqueeze(0)
53
 
54
  speech = model.generate_speech(input_ids, speaker_embedding, vocoder=vocoder)
55