Update goai_stt.py

goai_stt.py

@@ -1,4 +1,5 @@
 import torch
+import numpy as np
 import librosa
 import time
 from transformers import set_seed, Wav2Vec2ForCTC, AutoProcessor
@@ -37,11 +38,19 @@ def goai_stt(fichier):
     if isinstance(fichier, str):
         ### preprocessing de l'audio à partir d'un fichier
         signal, sampling_rate = librosa.load(fichier, sr=16000)
-    else:
+        
+    elif isinstance(fichier, np.ndarray):
         ### preprocessing de l'audio à partir d'un tableau numpy
-        signal, sampling_rate = fichier
+        signal = fichier
+        sampling_rate = 16000
+    else:
+        raise ValueError("Le type de fichier audio est invalide.")
 
-    inputs = processor(signal, sampling_rate=16_000, return_tensors="pt", padding=True).to(device)
+    # Ensure the signal is a numpy array and has correct dimensions
+    if not isinstance(signal, np.ndarray):
+        raise TypeError("Le signal audio doit être un tableau numpy.")
+    
+    inputs = processor(signal, sampling_rate=sampling_rate, return_tensors="pt", padding=True).to(device)
     
     ### faire l'inference
     with torch.no_grad():
@@ -51,4 +60,4 @@ def goai_stt(fichier):
     transcription = processor.decode(pred_ids)
 
     print("Temps écoulé: ", int(time.time() - start_time), " secondes")
-    return transcription
+    return transcription