fix: update output dir default path (#10)

* docs: update docs to point at docker hub * fix bug in default output dir
jveldboom · May 17, 2023 · 1611df8 · 1611df8
1 parent c797350
commit 1611df8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -19,11 +19,15 @@ Introducing Video Swear Jar, your AI-powered solution for creating clean video c
 
 ## Usage
 ```sh
-# build docker container
-make docker-build
-
 # create new video file with profanity removed
-docker run --rm -it -v $(pwd):/data video-swear-jar \
+docker run --rm -it \
+  -v $(pwd):/data jveldboom/video-swear-jar:v1 \
+  clean --input video.mkv --model tiny.en --language en
+
+# recommended to mount a ".whisper" directory to locally cache large language models
+docker run --rm -it \
+  -v $(pwd):/data \
+  -v $(pwd)/.whisper:/app/.whisper jveldboom/video-swear-jar:v1 \
   clean --input video.mkv --model tiny.en --language en
 ```
 

diff --git a/src/clean.js b/src/clean.js
@@ -24,7 +24,8 @@ const argv = yargs.usage('clean')
     },
     'output-dir': {
       description: 'Output directory to save cleaned video',
-      alias: 'o'
+      alias: 'o',
+      default: '.'
     }
   }).argv
 
@@ -34,7 +35,7 @@ const run = async () => {
   try {
     console.log(chalk.cyan('[1 of 4] Starting video transcribe...'))
     const { model, language } = argv
-    await video.transcribe({ inputFile: paths.inputFile, model, language, outputDir: argv.outputDir })
+    await video.transcribe({ inputFile: paths.inputFile, model, language, outputDir: argv['output-dir'] })
   } catch (err) {
     console.error(chalk.red(`Unable to transcribe ${paths.inputFile}`), err)
     throw err