File size: 608 Bytes
511e57c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import streamlit as st
from transformers import pipeline
if "classifier" not in st.session_state:
st.session_state["classifier"] = pipeline("text-classification", "Intel/polite-guard")
st.text_input("Your text here:", key="name")
# st.session_state.name
if "result" not in st.session_state:
st.session_state["result"] = "No result yet"
def click_button():
output = st.session_state["classifier"](st.session_state.name)
print(output)
st.session_state["result"] = output[0]["label"]
st.button('Run', on_click=click_button)
text1 = st.text_area('Result: ', st.session_state["result"])
|