summaryrefslogtreecommitdiffstats
path: root/roles/caption/templates/caption.sh
blob: 9600a3c97eefd6d7289e89d8537b51496eecf678 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
#!/bin/bash
# {{ ansible_managed }}
FILE="$1"
MODEL="${2:small}"
AUDIO=$(basename "$FILE" | sed s/\\.[a-z][a-z][a-z][a-z]?$//).ogg
if [[ ! -f $AUDIO ]]; then
  if  [[ "$FILE" == *webm ]]; then
      ffmpeg -y -i "$FILE" -acodec copy -vn $AUDIO
  else
      ffmpeg -y -i "$FILE" -acodec libvorbis -vn $AUDIO
  fi
fi
date > $AUDIO-$MODEL.log
time whisper $AUDIO --model $MODEL --threads 12 >> $AUDIO-$MODEL.log
for EXT in vtt txt srt; do
    mv $AUDIO.$EXT $(basename -s .webm.$EXT $AUDIO.$EXT)
done
date >> $AUDIO-$MODEL.log