anyantudre commited on
Commit
cfc8501
1 Parent(s): 0ae08c7

Update goai_tts.py

Browse files
Files changed (1) hide show
  1. goai_tts.py +4 -1
goai_tts.py CHANGED
@@ -4,7 +4,7 @@ import time
4
  from transformers import set_seed, pipeline
5
 
6
 
7
- def goai_tts(texte):
8
  """
9
  Pour un texte donné, donner le speech en Mooré correspondant
10
 
@@ -12,6 +12,9 @@ def goai_tts(texte):
12
  ----------
13
  texte: str
14
  Le texte écrit.
 
 
 
15
 
16
  Return
17
  ------
 
4
  from transformers import set_seed, pipeline
5
 
6
 
7
+ def goai_tts(texte, device):
8
  """
9
  Pour un texte donné, donner le speech en Mooré correspondant
10
 
 
12
  ----------
13
  texte: str
14
  Le texte écrit.
15
+
16
+ device: str
17
+ GPU ou CPU
18
 
19
  Return
20
  ------