feat(user/transcribe_whisper.sh):Specify dir_in via arg

[BK-2020-03.git] / user / transcribe_whisper.sh

diff --git a/user/transcribe_whisper.sh b/user/transcribe_whisper.sh
index 68756c74085f9fb88a65fc04effd328932940035..39944cea1bd6065d008327b5f6335ab4c59c5bc1 100755 (executable)
--- a/user/transcribe_whisper.sh
+++ b/user/transcribe_whisper.sh
@@ -1,8 +1,9 @@
 #!/bin/bash
 # Desc: Runs OpenAI Whisper on working directory media files
-# Usage: ./transcribe_whisper.sh 3
-# Input: arg1  CUDA graphics card number (zero-indexed)
-# Version: 0.1.0
+# Usage: ./transcribe_whisper.sh [dir] 3
+# Input: arg1  input dir
+#        arg2  CUDA graphics card number (zero-indexed)
+# Version: 0.2.0
 # Depends: whisper ( https://github.com/openai/whisper )
 
 # Find settings
@@ -52,11 +53,13 @@ find_flist() {
     must find "$1" -maxdepth "$fdepth" -type f -iregex "$firegex" -size +"$fsize";
 }; # print file list to stdout from dir with script parameters
 main() {
-    # Input: var:  fdepth   (find_flist) find depth
-    #        var:  firegex  (find_flist) pattern find iregex
-    #        var:  fsize    (find_flist) find size
-    cuda_num="$1";
-    dir_in="$(pwd)";
+    # Input: arg1: dir_in    input dir
+    #        arg2  cuda_num  cuda GPU index
+    #        var:  fdepth    (find_flist) find depth
+    #        var:  firegex   (find_flist) pattern find iregex
+    #        var:  fsize     (find_flist) find size
+    dir_in="$1";
+    cuda_num="$2";
     if ! checkInt "$cuda_num"; then die "FATAL:No graphics card selected."; fi;
     while read -r line; do
         echo "STATUS:Processing:$line" 1>&2;