From 0a34202ac70b2ed0f61343b74344633490edff81 Mon Sep 17 00:00:00 2001
From: Andres Marafioti <andimarafioti@gmail.com>
Date: Wed, 28 Aug 2024 16:18:16 +0200
Subject: [PATCH] edge case

---
 TTS/chatTTS_handler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/TTS/chatTTS_handler.py b/TTS/chatTTS_handler.py
index ee8ca25..6bdc6bf 100644
--- a/TTS/chatTTS_handler.py
+++ b/TTS/chatTTS_handler.py
@@ -58,7 +58,7 @@ class ChatTTSHandler(BaseHandler):
         if self.stream:
             wavs = [np.array([])]
             for gen in wavs_gen:
-                if len(gen[0]) == 0:
+                if gen[0] is None or len(gen[0]) == 0:
                     self.should_listen.set()
                     return
                 audio_chunk = librosa.resample(gen[0], orig_sr=24000, target_sr=16000)
-- 
GitLab