File size: 764 Bytes
f4dbd54 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import streamlit as st
from speechbrain.inference.interfaces import foreign_class
# Initialize the classifier
classifier = foreign_class(source="speechbrain/emotion-recognition-wav2vec2-IEMOCAP", pymodule_file="custom_interface.py", classname="CustomEncoderWav2vec2Classifier")
def emotion(file):
if file is not None:
# Classify the file
out_prob, score, index, text_lab = classifier.classify_file(file.name)
# Display the output
st.write(text_lab)
else:
st.write("Please upload a file.")
def main():
st.title("Emotion Recognition")
uploaded_file = st.file_uploader("Upload audio file", type=["wav"])
if uploaded_file is not None:
emotion(uploaded_file)
if __name__ == "__main__":
main()
|