Freddolin commited on
Commit
68f7e32
·
verified ·
1 Parent(s): ee62c26

Create tools/asr_tool.py

Browse files
Files changed (1) hide show
  1. tools/asr_tool.py +8 -0
tools/asr_tool.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ from faster_whisper import WhisperModel
2
+ import os
3
+
4
+ model = WhisperModel("small", device="cpu", compute_type="int8")
5
+
6
+ def transcribe_audio(audio_path: str) -> str:
7
+ segments, _ = model.transcribe(audio_path)
8
+ return " ".join(segment.text for segment in segments)