speech-to-text / app.py
pratikshahp's picture
Update app.py
7b7b977 verified
raw
history blame
385 Bytes
import librosa
import torch
from transformers import Wav2Vec2ForCTC, Wav2Vec2Tokenizer
import streamlit as st
#load pre-trained model and tokenizer
tokenizer = Wav2Vec2Tokenizer.from_pretrained("facebook/wav2vec2-base-960h")
model = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-base-960h")
#load audio file
speech, rate = librosa.load("batman1.wav",sr=16000)
st.write("hi")