Update goai_tts.py

goai_tts.py

@@ -3,8 +3,10 @@ import scipy
 import time
 from transformers import set_seed, pipeline
 
+device = 0 if torch.cuda.is_available() else "cpu"
 
-def goai_tts(texte, device):
+
+def goai_tts(texte):
     """
     Pour un texte donné, donner le speech en Mooré correspondant
     
@@ -12,9 +14,6 @@ def goai_tts(texte, device):
     ---------- 
     texte: str
         Le texte écrit.
-
-    device: str
-        GPU ou CPU
         
     Return
     ------