fixed bug. now output format is passed correctly
This commit is contained in:
parent
572d9a4398
commit
290052bfbf
15
main.sh
15
main.sh
@ -65,18 +65,6 @@ lower_ext() {
|
|||||||
|
|
||||||
is_url() { [[ "$1" =~ ^https?:// ]]; }
|
is_url() { [[ "$1" =~ ^https?:// ]]; }
|
||||||
|
|
||||||
output_flags_for() {
|
|
||||||
case "$1" in
|
|
||||||
txt) echo "--output-txt" ;;
|
|
||||||
vtt) echo "--output-vtt" ;;
|
|
||||||
srt) echo "--output-srt" ;;
|
|
||||||
json) echo "--output-json" ;;
|
|
||||||
tsv) echo "--output-csv" ;; # whisper.cpp uses CSV/TSV-ish; adjust if your build differs
|
|
||||||
lrc) echo "--output-lrc" ;;
|
|
||||||
all) echo "--output-txt --output-vtt --output-srt --output-json --output-csv --output-lrc" ;;
|
|
||||||
*) echo "Invalid --output_format: $1. Use txt|vtt|srt|json|tsv|lrc|all" >&2; exit 1 ;;
|
|
||||||
esac
|
|
||||||
}
|
|
||||||
|
|
||||||
# -------- Temp workspace & cleanup --------
|
# -------- Temp workspace & cleanup --------
|
||||||
WORKDIR="$(mktemp -d)"
|
WORKDIR="$(mktemp -d)"
|
||||||
@ -118,11 +106,10 @@ case "$ext" in
|
|||||||
esac
|
esac
|
||||||
|
|
||||||
# -------- Run whisper --------
|
# -------- Run whisper --------
|
||||||
OUT_FLAGS="$(output_flags_for "$OUTPUT_FORMAT")"
|
|
||||||
echo "▶ Running whisper"
|
echo "▶ Running whisper"
|
||||||
echo " model: $MODEL"
|
echo " model: $MODEL"
|
||||||
echo " language: $LANGUAGE"
|
echo " language: $LANGUAGE"
|
||||||
echo " outputs: $OUTPUT_FORMAT"
|
echo " outputs: $OUTPUT_FORMAT"
|
||||||
echo " input: $MP3_PATH"
|
echo " input: $MP3_PATH"
|
||||||
# shellcheck disable=SC2086
|
# shellcheck disable=SC2086
|
||||||
"$WHISPER_EXEC" "$MP3_PATH" --model "$MODEL" --device cuda --language "$LANGUAGE" $OUT_FLAGS
|
"$WHISPER_EXEC" "$MP3_PATH" --model "$MODEL" --device cuda --language "$LANGUAGE" --output_format $OUTPUT_FORMAT
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user