Update app.py

app.py

@@ -37,10 +37,8 @@ async def generate1(prompt):
     output = ""
     for response in stream:
         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 = output.strip('</s>')
+        yield output
 
     communicate = edge_tts.Communicate(output)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
@@ -66,10 +64,8 @@ async def generate2(prompt):
     output = ""
     for response in stream:
         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 = output.strip('</s>')
+        yield output
 
     communicate = edge_tts.Communicate(output)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
@@ -95,10 +91,8 @@ async def generate3(prompt):
     output = ""
     for response in stream:
         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 = output.strip('</s>')
+        yield output
 
     communicate = edge_tts.Communicate(output)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
@@ -116,7 +110,7 @@ with gr.Blocks(css="style.css") as demo:
                                 autoplay=True,
                                 elem_classes="audio")
     with gr.Row():
-        translate_btn = gr.Button("Generate")
+        translate_btn = gr.Button("Response")
         translate_btn.click(fn=generate1, inputs=user_input,
                             outputs=output_audio, api_name="translate")