Update goai_stt.py

goai_stt.py

@@ -3,8 +3,10 @@ import torch
 import time
 from transformers import set_seed, Wav2Vec2ForCTC, AutoProcessor
 
+device = 0 if torch.cuda.is_available() else "cpu"
 
-def goai_stt(fichier, device):
+
+def goai_stt(fichier):
     """
     Transcrire un fichier audio donné.