Merlintxu commited on
Commit
174a8ef
verified
1 Parent(s): 43399e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -6
app.py CHANGED
@@ -1,13 +1,22 @@
 
 
1
  from pyannote.audio import Pipeline
2
  import gradio as gr
3
 
4
  # Cargamos el pipeline preentrenado para diarizaci贸n
5
- pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization")
 
 
 
 
6
 
7
  # Funci贸n para realizar la diarizaci贸n de un archivo de audio
8
  def diarize_audio(file_path):
 
 
 
9
  # Realizar la diarizaci贸n del audio
10
- diarization = pipeline(file_path.name)
11
 
12
  # Crear un resultado legible con los segmentos de cada hablante
13
  result = ""
@@ -19,10 +28,10 @@ def diarize_audio(file_path):
19
  return result
20
 
21
  # Interfaz de Gradio
 
 
 
22
  audio_input = gr.Audio(type="file", label="Sube tu archivo de audio")
23
  text_output = gr.Textbox(label="Resultados de la Diarizaci贸n")
24
 
25
- gr.Interface(fn=diarize_audio, inputs=audio_input, outputs=text_output,
26
- title="Diarizaci贸n de Audio",
27
- description="Sube un archivo de audio para obtener los segmentos de los diferentes hablantes."
28
- ).launch()
 
1
+ # C贸digo para diarizaci贸n de audio
2
+ import os
3
  from pyannote.audio import Pipeline
4
  import gradio as gr
5
 
6
  # Cargamos el pipeline preentrenado para diarizaci贸n
7
+ try:
8
+ pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization")
9
+ except Exception as e:
10
+ pipeline = None
11
+ print(f"Error cargando el pipeline: {e}")
12
 
13
  # Funci贸n para realizar la diarizaci贸n de un archivo de audio
14
  def diarize_audio(file_path):
15
+ if pipeline is None:
16
+ return "Error: El pipeline no se pudo cargar. Verifica las dependencias."
17
+
18
  # Realizar la diarizaci贸n del audio
19
+ diarization = pipeline(file_path)
20
 
21
  # Crear un resultado legible con los segmentos de cada hablante
22
  result = ""
 
28
  return result
29
 
30
  # Interfaz de Gradio
31
+ def diarization_interface(file):
32
+ return diarize_audio(file.name)
33
+
34
  audio_input = gr.Audio(type="file", label="Sube tu archivo de audio")
35
  text_output = gr.Textbox(label="Resultados de la Diarizaci贸n")
36
 
37
+ gr.Interface(fn=diarization_interface, inputs=audio_input, outputs=text_output, title="Diarizaci贸n de Audio", description="Sube un archivo de audio para obtener los segmentos de los diferentes hablantes.").launch()