From 393600ad94c6bb244f0c8e2debc8cbb48f4c5df7 Mon Sep 17 00:00:00 2001
From: Paillat <git@electronique.cc>
Date: Mon, 15 May 2023 12:32:14 +0200
Subject: [PATCH] fix(speak.py): handle TTS initialization error when gpu is
 not available

---
 generators/speak.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/generators/speak.py b/generators/speak.py
index da96855..95b49ab 100644
--- a/generators/speak.py
+++ b/generators/speak.py
@@ -1,7 +1,5 @@
 from TTS.api import TTS
 
-import os
-
 # Running a multi-speaker and multi-lingual model
 
 # List available 🐸TTS models and choose the first one
@@ -18,6 +16,9 @@ voices = ["Alexander", "Benjamin", "Amelia", "Katherine"]
 # Init TTS
 
 def generate_voice(path, text, speaker="Alexander"):
-    tts = TTS(model_best_multi, gpu=True)
+    try:
+        tts = TTS(model_best_multi, gpu=True)
+    except:
+        tts = TTS(model_best_multi, gpu=False)
     speaker = fakenames[speaker] if speaker in fakenames else speaker
     tts.tts_to_file(text=text, file_path=path, speaker=speaker, speed=1)
\ No newline at end of file