andre-coy commited on
Commit
571cd0d
โ€ข
1 Parent(s): 03301c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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["input_ids"], speaker_embedding, vocoder=vocoder)
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(inputs["input_ids"], speaker_embedding, vocoder=vocoder)
56
 
57
  speech = (speech.numpy() * 32767).astype(np.int16)
58
  return (16000, speech)