Kr08 commited on
Commit
adc7ace
·
verified ·
1 Parent(s): 9c818ec

Update audio_processing.py

Browse files
Files changed (1) hide show
  1. audio_processing.py +3 -1
audio_processing.py CHANGED
@@ -32,7 +32,7 @@ def preprocess_audio(audio, chunk_size=CHUNK_LENGTH*16000, overlap=OVERLAP*16000
32
  chunks.append(chunk)
33
  return chunks
34
 
35
- @spaces.GPU()
36
  def process_audio(audio_file, translate=False, model_size="small"):
37
  start_time = time.time()
38
 
@@ -113,6 +113,7 @@ def process_audio(audio_file, translate=False, model_size="small"):
113
  logger.error(f"An error occurred during audio processing: {str(e)}")
114
  raise
115
 
 
116
  def merge_nearby_segments(segments, time_threshold=0.5, similarity_threshold=0.9):
117
  merged = []
118
  for segment in segments:
@@ -136,6 +137,7 @@ def merge_nearby_segments(segments, time_threshold=0.5, similarity_threshold=0.9
136
  merged.append(segment)
137
  return merged
138
 
 
139
  def print_results(segments):
140
  for segment in segments:
141
  print(f"[{segment['start']:.2f}s - {segment['end']:.2f}s] ({segment['language']}) {segment['speaker']}:")
 
32
  chunks.append(chunk)
33
  return chunks
34
 
35
+ @spaces.GPU(duration=60)
36
  def process_audio(audio_file, translate=False, model_size="small"):
37
  start_time = time.time()
38
 
 
113
  logger.error(f"An error occurred during audio processing: {str(e)}")
114
  raise
115
 
116
+
117
  def merge_nearby_segments(segments, time_threshold=0.5, similarity_threshold=0.9):
118
  merged = []
119
  for segment in segments:
 
137
  merged.append(segment)
138
  return merged
139
 
140
+
141
  def print_results(segments):
142
  for segment in segments:
143
  print(f"[{segment['start']:.2f}s - {segment['end']:.2f}s] ({segment['language']}) {segment['speaker']}:")