andre-coy commited on
Commit
7d98f33
โ€ข
1 Parent(s): a361a36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ def predict(text, speaker):
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
 
56
  speech = (speech.numpy() * 32767).astype(np.int16)
 
50
  speaker_embedding = np.load(speaker_embeddings[speaker[:3]])
51
 
52
  #speaker_embedding = torch.tensor(speaker_embedding).unsqueeze(0)
53
+ speaker_embedding = torch.tensor(speaker_embedding) #the saved model is already unsqueezed, but is not a tensor, so make it one
54
  speech = model.generate_speech(input_ids, speaker_embedding, vocoder=vocoder)
55
 
56
  speech = (speech.numpy() * 32767).astype(np.int16)