Added time to output and default model value

app.py

@@ -2,11 +2,15 @@ import gradio as gr
 import torch
 from transformers import pipeline
 import numpy as np
+import time
 
 pipe_base = pipeline("automatic-speech-recognition", model="aitor-medrano/lara-base-pushed")
 pipe_small = pipeline("automatic-speech-recognition", model="aitor-medrano/whisper-small-lara")
 
-def greet(modelo, grabacion):
+def greet(modelo="Base", grabacion):
+
+    inicio = time.time()
+
     sr, y = grabacion
     # Pasamos el array de muestras a tipo NumPy de 32 bits
     y = y.astype(np.float32)
@@ -17,14 +21,17 @@ def greet(modelo, grabacion):
     else:
         pipe = pipe_small
 
-    return modelo + ":" + pipe({"sampling_rate": sr, "raw": y})["text"]
+    result = modelo + ":" + pipe({"sampling_rate": sr, "raw": y})["text"]
+    fin = time.time()
+
+    return result, fin - inicio
 
 demo = gr.Interface(fn=greet,
-                    inputs=[
-                            gr.Dropdown(
-                                ["base", "small"], label="Modelo", info="Modelos de Lara entrenados"
-                            ),
-                            gr.Audio()
-                    ],
-                    outputs="text")
+        inputs=[
+                gr.Dropdown(
+                    ["base", "small"], label="Modelo", info="Modelos de Lara entrenados"
+                ),
+                gr.Audio()
+        ],
+        outputs=["text","number"])
 demo.launch()