refactor transcribe-voice-track.py

This commit is contained in:
2021-12-13 19:02:28 -07:00
parent f090e13655
commit 4138dbf353
3 changed files with 17 additions and 4 deletions

3
scripts/requirements.txt Normal file
View File

@@ -0,0 +1,3 @@
requests
vosk
scipy

View File

@@ -1,6 +1,7 @@
# pip install requests
# pip install volk
# pip install scipy
# pip install -r requirements.txt
usage = 'python transcribe-voice-track.py <?wav filename>'
import util
import wave
import json
import sys
@@ -27,7 +28,7 @@ if not os.path.exists(model_path):
with ZipFile(model_zip_path, "r") as zip_file:
zip_file.extractall('models')
audio_filename = sys.argv[1] if len(sys.argv) > 1 else input("mono-track wav filename? ")
audio_filename = util.arg(1, usage)
wf = wave.open(audio_filename, "rb")
model = Model(model_path)

9
scripts/util.py Normal file
View File

@@ -0,0 +1,9 @@
import sys
def arg(num, usage):
val = ''
if len(sys.argv) > num:
val = sys.argv[num]
else:
raise ValueError(usage)
return val