Update app.py

app.py

@@ -195,13 +195,13 @@ def tts_streaming(text: str, voice: str = "af_heart", speed: float = 1.0, format
 
             # For the first chunk, prepend 0; for later chunks, start with the previous chunk's last token.
             if i == 0:
-                tokens_to_send = [0] + chunk_tokens
+                tokens_to_send = [0] + chunk_tokens + [0]
             else:
-                tokens_to_send =  [0] + prev_last_token + [16] + chunk_tokens
+                tokens_to_send =  [0] + prev_last_token + [16] + chunk_tokens + [0]
                 # token_to_send = [0] + chunk_tokens
 
             # Save the last token of this chunk for the next iteration.
-            prev_last_token = chunk_tokens[-2:]
+            prev_last_token = chunk_tokens[-1:]
 
             # Prepare the model input (a batch of one sequence).
             final_token = [tokens_to_send]