papasega commited on
Commit
fe9b085
1 Parent(s): b691b31

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -6
app.py CHANGED
@@ -24,14 +24,18 @@ import numpy as np
24
  # Charger le modèle pré-entraîné
25
  model = EncoderASR.from_hparams("speechbrain/asr-wav2vec2-dvoice-wolof")
26
 
27
- # Définir la fonction de transcription
 
 
 
 
 
 
 
 
28
  def transcribe(audio):
29
- sr, y = audio
30
- y = y.astype(np.float32)
31
- y /= np.max(np.abs(y))
32
 
33
- # Utiliser le modèle pour transcrire l'audio
34
- return model.transcribe({"sampling_rate": sr, "raw": y})["text"]
35
 
36
  # Créer l'interface Gradio avec le microphone et le téléchargement de fichier comme options d'entrée
37
  asrdemo = gr.Interface(
 
24
  # Charger le modèle pré-entraîné
25
  model = EncoderASR.from_hparams("speechbrain/asr-wav2vec2-dvoice-wolof")
26
 
27
+ # # Définir la fonction de transcription
28
+ # def transcribe(audio):
29
+ # sr, y = audio
30
+ # y = y.astype(np.float32)
31
+ # y /= np.max(np.abs(y))
32
+
33
+ # # Utiliser le modèle pour transcrire l'audio
34
+ # return model.transcribe({"sampling_rate": sr, "raw": y})["text"]
35
+
36
  def transcribe(audio):
37
+ return model.transcribe_file(audio.name)
 
 
38
 
 
 
39
 
40
  # Créer l'interface Gradio avec le microphone et le téléchargement de fichier comme options d'entrée
41
  asrdemo = gr.Interface(