Spaces:
Running
on
Zero
Running
on
Zero
Update audio_processing.py
Browse files- 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']}:")
|