Update app.py

app.py

@@ -37,8 +37,12 @@ async def generate1(prompt):
     output = ""
     for response in stream:
         output += response.token.text
-        output = output.strip('</s>')
-        yield output
+
+    # Clean the output to remove extraneous characters
+    output = re.sub(r'[\s/]+', ' ', output).strip()
+    output = re.sub(r'\s*$', '', output).strip()  # Remove trailing whitespaces
+    if output.endswith(' s'):
+        output = output[:-2].strip()  # Remove trailing ' s'
 
     communicate = edge_tts.Communicate(output)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
@@ -64,8 +68,12 @@ async def generate2(prompt):
     output = ""
     for response in stream:
         output += response.token.text
-        output = output.strip('</s>')
-        yield output
+
+    # Clean the output to remove extraneous characters
+    output = re.sub(r'[\s/]+', ' ', output).strip()
+    output = re.sub(r'\s*$', '', output).strip()  # Remove trailing whitespaces
+    if output.endswith(' s'):
+        output = output[:-2].strip()  # Remove trailing ' s'
 
     communicate = edge_tts.Communicate(output)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
@@ -91,8 +99,12 @@ async def generate3(prompt):
     output = ""
     for response in stream:
         output += response.token.text
-        output = output.strip('</s>')
-        yield output
+
+    # Clean the output to remove extraneous characters
+    output = re.sub(r'[\s/]+', ' ', output).strip()
+    output = re.sub(r'\s*$', '', output).strip()  # Remove trailing whitespaces
+    if output.endswith(' s'):
+        output = output[:-2].strip()  # Remove trailing ' s'
 
     communicate = edge_tts.Communicate(output)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file: