Update utils_hf.py

utils_hf.py

@@ -128,10 +128,8 @@ def extract_audio_segment(input_audio_path, output_audio_path, start_time, end_t
 def transcribe_audio(audio_path):
     """
     Transcribe el contenido de un archivo de audio utilizando el modelo Whisper de Hugging Face.
-
     Args:
         audio_path (str): Ruta del archivo de audio a transcribir.
-
     Returns:
         str: Texto transcrito del audio.
     """
@@ -144,13 +142,12 @@ def transcribe_audio(audio_path):
             audio_data = np.mean(audio_data, axis=1)
 
         # Transcribir el audio usando el modelo Whisper
-        transcript = whisper_pipeline(audio_data)
+        transcript = whisper_pipeline(audio_data, return_timestamps=True)
         return transcript['text']
     except Exception as e:
         st.error(f"Error al transcribir el audio: {e}")
         st.stop()
 
-
 def get_audio_duration(audio_path):
     """
     Obtiene la duración de un archivo de audio.