Etiquetas de salida

app.py

@@ -7,7 +7,7 @@ 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(grabacion, modelo="Base"):
+def greet(grabacion, modelo="base"):
 
     inicio = time.time()
 
@@ -16,7 +16,7 @@ def greet(grabacion, modelo="Base"):
     y = y.astype(np.float32)
     y /= np.max(np.abs(y))
 
-    if modelo == "Base":
+    if modelo == "base":
         pipe = pipe_base
     else:
         pipe = pipe_small
@@ -33,5 +33,8 @@ demo = gr.Interface(fn=greet,
                     ["base", "small"], label="Modelo", info="Modelos de Lara entrenados"
                 )
         ],
-        outputs=["text","number"])
+        outputs=[
+            gr.Text(label="Salida"),
+            gr.Number(label="Tiempo")
+        ])
 demo.launch()