Martin Bär commited on
Commit
482bc3b
·
1 Parent(s): b85582b

Add check if audio is long

Browse files
Files changed (1) hide show
  1. multimodality_tools.py +4 -1
multimodality_tools.py CHANGED
@@ -27,7 +27,10 @@ def transcribe_audio(file_id: str) -> str:
27
 
28
  asr = pipeline("automatic-speech-recognition", model="openai/whisper-tiny")
29
 
30
- output = asr(audio)
 
 
 
31
 
32
  return output["text"].strip()
33
 
 
27
 
28
  asr = pipeline("automatic-speech-recognition", model="openai/whisper-tiny")
29
 
30
+ if (len(audio) / 16000) > 25:
31
+ output = asr(audio, return_timestamps=True)
32
+ else:
33
+ output = asr(audio)
34
 
35
  return output["text"].strip()
36