mgruner's picture
Add simple script to run a local transcription
fa1dce6
raw
history blame
295 Bytes
import sys
import nemo.collections.asr as nemo_asr
def main(argv):
model = nemo_asr.models.ASRModel.from_pretrained(
model_name="nvidia/parakeet-tdt-0.6b-v2"
)
output = model.transcribe([argv[1]])
print(output[0].text)
if __name__ == "__main__":
main(sys.argv)