wav2vec-stt / usage.py
datasetsANDmodels's picture
Rename wav2vec_stt.py to usage.py
8ec1e69 verified
import sys
from transformers import pipeline
model="wav2vec-stt"
pipe = pipeline("automatic-speech-recognition", model=model)
audio_file = sys.argv[1]
result = pipe(audio_file)["text"]
print(result)