CarPeAs commited on
Commit
378c8a1
·
1 Parent(s): 7afb5e2

añadiendo temporizador

Browse files
Files changed (2) hide show
  1. __pycache__/app.cpython-39.pyc +0 -0
  2. app.py +10 -2
__pycache__/app.cpython-39.pyc CHANGED
Binary files a/__pycache__/app.cpython-39.pyc and b/__pycache__/app.cpython-39.pyc differ
 
app.py CHANGED
@@ -2,6 +2,7 @@ from transformers import pipeline
2
  import gradio as gr
3
  import numpy as np
4
  import librosa
 
5
 
6
  # Utilizamos los tres modelos entrenados
7
  pipe_model_1 = pipeline("automatic-speech-recognition", model="IABDs8a/AfinandoElEntrenamiento")
@@ -9,6 +10,8 @@ pipe_model_2 = pipeline("automatic-speech-recognition", model="IABDs8a/whisper-b
9
  pipe_model_3 = pipeline("automatic-speech-recognition", model="IABDs8a/whisper-tiny-top3")
10
 
11
  def transcribe(audio, model_choice):
 
 
12
  if model_choice == "AfinandoElEntrenamiento":
13
  pipe = pipe_model_1
14
  elif model_choice == "Whisper Base Full":
@@ -31,7 +34,9 @@ def transcribe(audio, model_choice):
31
 
32
  # Realizar la transcripción
33
  result = pipe({"sampling_rate": sr, "raw": y})
34
- return result["text"]
 
 
35
 
36
  # Interfaz de Gradio
37
  demo = gr.Interface(
@@ -40,7 +45,10 @@ demo = gr.Interface(
40
  gr.Audio(type="filepath", label="Sube un archivo de audio o graba desde el micrófono"),
41
  gr.Dropdown(choices=["AfinandoElEntrenamiento", "Whisper Base Full", "Whisper Tiny Top 3"], label="Selecciona el modelo", value="Whisper Base Full")
42
  ],
43
- outputs="text",
 
 
 
44
  title="Transcripción de Audio con LARA",
45
  description="Sube un archivo de audio o graba desde el micrófono para obtener su transcripción utilizando los modelos Whisper entrenados.",
46
  )
 
2
  import gradio as gr
3
  import numpy as np
4
  import librosa
5
+ import time
6
 
7
  # Utilizamos los tres modelos entrenados
8
  pipe_model_1 = pipeline("automatic-speech-recognition", model="IABDs8a/AfinandoElEntrenamiento")
 
10
  pipe_model_3 = pipeline("automatic-speech-recognition", model="IABDs8a/whisper-tiny-top3")
11
 
12
  def transcribe(audio, model_choice):
13
+ inicio = time.time()
14
+
15
  if model_choice == "AfinandoElEntrenamiento":
16
  pipe = pipe_model_1
17
  elif model_choice == "Whisper Base Full":
 
34
 
35
  # Realizar la transcripción
36
  result = pipe({"sampling_rate": sr, "raw": y})
37
+ fin = time.time()
38
+
39
+ return result["text"], fin - inicio
40
 
41
  # Interfaz de Gradio
42
  demo = gr.Interface(
 
45
  gr.Audio(type="filepath", label="Sube un archivo de audio o graba desde el micrófono"),
46
  gr.Dropdown(choices=["AfinandoElEntrenamiento", "Whisper Base Full", "Whisper Tiny Top 3"], label="Selecciona el modelo", value="Whisper Base Full")
47
  ],
48
+ outputs=[
49
+ gr.Text(label="Salida"),
50
+ gr.Number(label="Tiempo")
51
+ ],
52
  title="Transcripción de Audio con LARA",
53
  description="Sube un archivo de audio o graba desde el micrófono para obtener su transcripción utilizando los modelos Whisper entrenados.",
54
  )