anyantudre commited on
Commit
1d31b00
·
verified ·
1 Parent(s): ca6951f

Update goai_tts.py

Browse files
Files changed (1) hide show
  1. goai_tts.py +3 -0
goai_tts.py CHANGED
@@ -3,6 +3,9 @@ import numpy as np
3
  import scipy.io.wavfile
4
  from transformers import pipeline, set_seed
5
 
 
 
 
6
  def goai_tts(texte):
7
  """
8
  Pour un texte donné, donner le speech en Mooré correspondant
 
3
  import scipy.io.wavfile
4
  from transformers import pipeline, set_seed
5
 
6
+ device = 0 if torch.cuda.is_available() else "cpu"
7
+
8
+
9
  def goai_tts(texte):
10
  """
11
  Pour un texte donné, donner le speech en Mooré correspondant