Spaces:

broadfield
/

chat-voice

Sleeping

broadfield commited on Dec 4, 2024

Commit

75c35f4

verified ·

1 Parent(s): 1cb3b4a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ def generate(prompt,history,max_new_tokens,seed):
     output = ""
     buf = ""
     for response in stream:
-        if response.token.text == "\n":
             yield from pp.stream_tts(buf)
             buf=""
         buf += response.token.text

     output = ""
     buf = ""
     for response in stream:
+        if response.token.text == "\n" or response.token.text == "</s>":
             yield from pp.stream_tts(buf)
             buf=""
         buf += response.token.text