andre-coy commited on
Commit
a799b36
·
verified ·
1 Parent(s): 9fbb341

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -54,7 +54,9 @@ def predict(text, speaker):
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)
59
 
60
 
 
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
+ speech = np.tanh(speech) * 32767 # Non-linear scaling to reduce harshness
59
+ speech = speech.astype(np.int16)
60
  return (16000, speech)
61
 
62