fix re sub bug (infiniflow#2199)

### What problem does this PR solve? ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue)
Halfknow · Sep 2, 2024 · 218c4c7 · 218c4c7
1 parent ac94521
commit 218c4c7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/rag/llm/tts_model.py b/rag/llm/tts_model.py
@@ -53,9 +53,10 @@ def __init__(self, key, model_name, base_url):
     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: