Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -52,7 +52,7 @@ def predict(text, speaker):
|
|
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 |
-
speech = model.generate_speech(input_ids["
|
56 |
|
57 |
speech = (speech.numpy() * 32767).astype(np.int16)
|
58 |
return (16000, speech)
|
|
|
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 |
+
speech = model.generate_speech(input_ids["input_ids"], speaker_embedding, vocoder=vocoder)
|
56 |
|
57 |
speech = (speech.numpy() * 32767).astype(np.int16)
|
58 |
return (16000, speech)
|