Update goai_tts.py

goai_tts.py

@@ -1,6 +1,7 @@
 import torch
 import scipy
 import time
+import numpy as np
 from transformers import set_seed, pipeline
 
 device = 0 if torch.cuda.is_available() else "cpu"
@@ -34,4 +35,5 @@ def goai_tts(texte):
     wavfile = scipy.io.wavfile.write("finetuned_output.wav", rate=speech["sampling_rate"], data=speech["audio"][0])
 
     print("Temps écoulé: ", int(time.time() - start_time), " secondes")
-    return numpy.array(wavfile[1], dtype=float)
+    
+    return np.array(wavfile[1], dtype=float)