| import streamlit as st | |
| from transformers import pipeline | |
| from datasets import load_dataset | |
| audio_classifier = pipeline( | |
| task="audio-classification", model="ehcalabres/wav2vec2-lg-xlsr-en-speech-emotion-recognition" | |
| def main(): | |
| st.title("Yelp review") | |
| with st.form("text_field"): | |
| text = st.text_area('enter some text:') | |
| # clicked==True only when the button is clicked | |
| clicked = st.form_submit_button("Submit") | |
| if clicked: | |
| results = classifier([text]) | |
| st.json(results) | |
| if __name__ == "__main__": | |
| main() |