Freddolin commited on
Commit
464e3d9
·
verified ·
1 Parent(s): 6cd4f68

Update tools/asr_tool.py

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