Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|