| from inference import infer | |
| import streamlit as st | |
| # | |
| # This is the app.py file required by streamlit (huggingface spaces) | |
| # See streamlit docs for more info https://docs.streamlit.io/ | |
| # Develop this app locally using: | |
| # streamlit run app.py | |
| # test in a browser: http://localhost:8501 | |
| # When ready to update the space, push this repo to huggingface: | |
| # ./push_to_space | |
| # | |
| def text_cb(): | |
| response = infer(st.session_state.inp_text) | |
| # st.write(response) | |
| for key, value in response.items(): | |
| if key != "neutral" or len(response.keys())==1: | |
| bottom.write(f"{key}: {value}") | |
| st.text_input("Type text in the box below", on_change=text_cb, key="inp_text") | |
| bottom = st.container() | |
| bottom.empty() |