Paillat-dev/viralfactory
1
0
Fork 0
mirror of https://github.com/Paillat-dev/viralfactory.git synced 2026-01-02 09:16:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Formatting and stuff

Browse Source
This commit is contained in:
Paillat
2024-02-23 11:07:50 +01:00
parent 8951837d04
commit d7fa8945d1
13 changed files with 28 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/engines/TTSEngine/BaseTTSEngine.py
View File

@@ -22,17 +22,6 @@ class BaseTTSEngine(BaseEngine):
def remove_punctuation(self, text: str) -> str:
return text.translate(str.maketrans("", "", ".,!?;:"))
def fix_captions(self, script: str, captions: list[Word]) -> list[Word]:
script = script.split(" ")
new_captions = []
for i, word in enumerate(script):
original_word = self.remove_punctuation(word.lower())
stt_word = self.remove_punctuation(word.lower())
if stt_word in original_word:
captions[i]["text"] = word
new_captions.append(captions[i])
# elif there is a word more in the stt than in the original, we
def time_with_whisper(self, path: str) -> list[Word]:
"""
Transcribes the audio file at the given path using a pre-trained model and returns a list of words.