aljaz/nostrdvm
1
0
Fork 0
mirror of https://github.com/aljazceru/nostrdvm.git synced 2026-01-06 07:44:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

some fixes for tts, adding experimental code for gallery lists

Browse Source
This commit is contained in:
Believethehype
2024-06-27 21:19:04 +02:00
parent ffcedafe74
commit f7e1bc5e52
7 changed files with 61 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
nostr_dvm/tasks/texttospeech.py
View File

@@ -125,15 +125,15 @@ class TextToSpeech(DVMTaskInterface):
model = "tts_models/multilingual/multi-dataset/xtts_v2"
tts = TTS(model).to(device)
text = options["prompt"]
text = options["prompt"].replace("~", "around ").replace("#", "hashtag").replace(" LFG", " Let's fucking go")
text_clean = re.sub(
r'''(?i)\b((?:https?://|www\d{0,3}[.]|[a-z0-9.\-]+[.][a-z]{2,4}/)(?:[^\s()<>]+|\(([^\s()<>]+|(\([^\s()<>]+\)))*\))+(?:\(([^\s()<>]+|(\([^\s()<>]+\)))*\)|[^\s`!()\[\]{};:'".,<>?«»“”‘’]))''',
" ", text)
tts.tts_to_file(
text=text_clean,
speaker_wav=options["input_wav"], language=options["language"], file_path="outputs/output.wav")
result = upload_media_to_hoster("outputs/output.wav")
speaker_wav=options["input_wav"], language=options["language"], file_path="outputs/output.mp3")
result = upload_media_to_hoster("outputs/output.mp3")
print(result)
return result
except Exception as e: