AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-20 00:07:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix elevenlabs_tts too

Browse Source
This commit is contained in:
oobabooga 2023-05-21 14:11:46 -03:00
parent 1e5821bd9e
commit 767a767989
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
extensions/elevenlabs_tts/script.py
View File

@ -71,17 +71,21 @@ def input_modifier(string):
This function is applied to your text inputs before This function is applied to your text inputs before
they are fed into the model. they are fed into the model.
""" """
shared.processing_message = "*Is recording a voice message...*"
return string
def history_modifier(history):
# Remove autoplay from the last reply # Remove autoplay from the last reply
if shared.is_chat() and len(shared.history['internal']) > 0: if len(history['internal']) > 0:
shared.history['visible'][-1] = [ history['visible'][-1] = [
shared.history['visible'][-1][0], history['visible'][-1][0],
shared.history['visible'][-1][1].replace('controls autoplay>', 'controls>') history['visible'][-1][1].replace('controls autoplay>', 'controls>')
] ]
if params['activate']: return history
shared.processing_message = "*Is recording a voice message...*"
return string
def output_modifier(string): def output_modifier(string):
@ -104,7 +108,7 @@ def output_modifier(string):
string = 'empty reply, try regenerating' string = 'empty reply, try regenerating'
output_file = Path(f'extensions/elevenlabs_tts/outputs/{wav_idx:06d}.mp3'.format(wav_idx)) output_file = Path(f'extensions/elevenlabs_tts/outputs/{wav_idx:06d}.mp3'.format(wav_idx))
print(f'Outputing audio to {str(output_file)}') print(f'Outputting audio to {str(output_file)}')
try: try:
audio = elevenlabs.generate(text=string, voice=params['selected_voice'], model="eleven_monolingual_v1") audio = elevenlabs.generate(text=string, voice=params['selected_voice'], model="eleven_monolingual_v1")
elevenlabs.save(audio, str(output_file)) elevenlabs.save(audio, str(output_file))