Update app.py

app.py

@@ -3,7 +3,6 @@ import re
 import gradio as gr
 import edge_tts
 import asyncio
-import time
 import tempfile
 from huggingface_hub import InferenceClient
 
@@ -40,7 +39,8 @@ async def generate1(prompt):
         output += response.token.text
 
     # Clean the output to remove extraneous characters
-    output = re.sub(r'\s+', ' ', output).strip()
+    output = re.sub(r'[\s/]+', ' ', output).strip()
+    output = re.sub(r'\/s$', '', output).strip()
 
     communicate = edge_tts.Communicate(output)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
@@ -68,7 +68,8 @@ async def generate2(prompt):
         output += response.token.text
 
     # Clean the output to remove extraneous characters
-    output = re.sub(r'\s+', ' ', output).strip()
+    output = re.sub(r'[\s/]+', ' ', output).strip()
+    output = re.sub(r'\/s$', '', output).strip()
 
     communicate = edge_tts.Communicate(output)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
@@ -96,7 +97,8 @@ async def generate3(prompt):
         output += response.token.text
 
     # Clean the output to remove extraneous characters
-    output = re.sub(r'\s+', ' ', output).strip()
+    output = re.sub(r'[\s/]+', ' ', output).strip()
+    output = re.sub(r'\/s$', '', output).strip()
 
     communicate = edge_tts.Communicate(output)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
@@ -114,7 +116,7 @@ with gr.Blocks(css="style.css") as demo:
                                 autoplay=True,
                                 elem_classes="audio")
     with gr.Row():
-        translate_btn = gr.Button("Response")
+        translate_btn = gr.Button("Generate")
         translate_btn.click(fn=generate1, inputs=user_input,
                             outputs=output_audio, api_name="translate")