refactor transcribe-voice-track.py
This commit is contained in:
3
projects/hollywoo/scripts/requirements.txt
Normal file
3
projects/hollywoo/scripts/requirements.txt
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
requests
|
||||||
|
vosk
|
||||||
|
scipy
|
||||||
@@ -1,6 +1,7 @@
|
|||||||
# pip install requests
|
# pip install -r requirements.txt
|
||||||
# pip install volk
|
usage = 'python transcribe-voice-track.py <?wav filename>'
|
||||||
# pip install scipy
|
|
||||||
|
import util
|
||||||
import wave
|
import wave
|
||||||
import json
|
import json
|
||||||
import sys
|
import sys
|
||||||
@@ -27,7 +28,7 @@ if not os.path.exists(model_path):
|
|||||||
with ZipFile(model_zip_path, "r") as zip_file:
|
with ZipFile(model_zip_path, "r") as zip_file:
|
||||||
zip_file.extractall('models')
|
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")
|
wf = wave.open(audio_filename, "rb")
|
||||||
|
|
||||||
model = Model(model_path)
|
model = Model(model_path)
|
||||||
|
|||||||
9
projects/hollywoo/scripts/util.py
Normal file
9
projects/hollywoo/scripts/util.py
Normal 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
|
||||||
Reference in New Issue
Block a user