search

dusty-nv / jetson-containers

Machine Learning Containers for NVIDIA Jetson and JetPack-L4T
MIT License
2.13k stars 440 forks source link

Add Whisper package #321

Closed tokk-nv closed 10 months ago

tokk-nv commented 10 months ago

OpenAI's Whisper package. Added an extra example Jupyter notebook, to use microphone to record audio and then transcribe.

image

johnnynunez commented 10 months ago

Can you use whisper-X? I have problems with that

tokk-nv commented 10 months ago

Hi @johnnynunez, Thank you for your suggestion! I added a whisperx package in my fork (and made a draft-PR #322), so it would be great if you can test it if you have a Jetson device.

You can

git clone --branch dev-whisperx https://github.com/tokk-nv/jetson-containers
cd jetson-containers
./build.sh whisperx --skip-test=all --no-github-api
./run.sh $(./autotag whisperx)

What I tested was, first coping a sample wav file in jetson-containers/data/audio/ directory, and then in the container;

whisperx /data/audio/Micro-Machines.wav
cat Micro-Machines.txt

I'm curious to know how other examples would work, so it would be great if you can share your test results. Thank you!