speechemotion / app.py
Nattyboi's picture
Update app.py
a97191b
raw
history blame
334 Bytes
import streamlit
# Load model directly
from transformers import AutoProcessor, Wav2Vec2ForSpeechClassification
processor = AutoProcessor.from_pretrained("harshit345/xlsr-wav2vec-speech-emotion-recognition")
model = Wav2Vec2ForSpeechClassification.from_pretrained("harshit345/xlsr-wav2vec-speech-emotion-recognition")
st.write(model)