Update goai_stt.py

goai_stt.py

@@ -4,7 +4,7 @@ import time
 from transformers import set_seed, Wav2Vec2ForCTC, AutoProcessor
 
 
-def goai_stt(fichier):
+def goai_stt(fichier, device):
     """
     Transcrire un fichier audio donné.
     
@@ -12,6 +12,9 @@ def goai_stt(fichier):
     ---------- 
     fichier: str
         Le chemin d'accès au fichier audio.
+
+    device: str
+        GPU ou CPU
         
     Return
     ----------