diff --git a/rag/llm/tts_model.py b/rag/llm/tts_model.py index 3ae170ee8..c477313a9 100644 --- a/rag/llm/tts_model.py +++ b/rag/llm/tts_model.py @@ -53,9 +53,10 @@ class Base(ABC): def tts(self, audio): pass - def normalize_text(text): + def normalize_text(self, text): return re.sub(r'(\*\*|##\d+\$\$|#)', '', text) + class FishAudioTTS(Base): def __init__(self, key, model_name, base_url="https://api.fish.audio/v1/tts"): if not base_url: