Spaces:
Configuration error
Configuration error
Fedir Zadniprovskyi
commited on
Commit
·
e41bc7f
1
Parent(s):
4bdd7f2
chore: log file transcription time
Browse files- speaches/main.py +6 -0
speaches/main.py
CHANGED
|
@@ -67,14 +67,20 @@ async def transcribe_file(
|
|
| 67 |
assert (
|
| 68 |
model == config.whisper.model
|
| 69 |
), "Specifying a model that is different from the default is not supported yet."
|
|
|
|
| 70 |
segments, transcription_info = whisper.transcribe(
|
| 71 |
file.file,
|
| 72 |
language=language,
|
| 73 |
initial_prompt=prompt,
|
| 74 |
word_timestamps="words" in timestamp_granularities,
|
| 75 |
temperature=temperature,
|
|
|
|
| 76 |
)
|
| 77 |
segments = list(segments)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 78 |
if response_format == ResponseFormat.TEXT:
|
| 79 |
return utils.segments_text(segments)
|
| 80 |
elif response_format == ResponseFormat.JSON:
|
|
|
|
| 67 |
assert (
|
| 68 |
model == config.whisper.model
|
| 69 |
), "Specifying a model that is different from the default is not supported yet."
|
| 70 |
+
start = time.perf_counter()
|
| 71 |
segments, transcription_info = whisper.transcribe(
|
| 72 |
file.file,
|
| 73 |
language=language,
|
| 74 |
initial_prompt=prompt,
|
| 75 |
word_timestamps="words" in timestamp_granularities,
|
| 76 |
temperature=temperature,
|
| 77 |
+
vad_filter=True,
|
| 78 |
)
|
| 79 |
segments = list(segments)
|
| 80 |
+
end = time.perf_counter()
|
| 81 |
+
logger.info(
|
| 82 |
+
f"Transcribed {transcription_info.duration}({transcription_info.duration_after_vad}) in {end - start:.2f} seconds"
|
| 83 |
+
)
|
| 84 |
if response_format == ResponseFormat.TEXT:
|
| 85 |
return utils.segments_text(segments)
|
| 86 |
elif response_format == ResponseFormat.JSON:
|