Zekun Wu
update
40c82a6
raw
history blame
1.06 kB
import streamlit as st
from bias_detector import Detector
st.title("Multidimensional Multilevel Bias Detection")
level = st.selectbox("Select the Bias Levels:", ("Token","Sentence"))
dimension = st.selectbox("Select the Bias Dimensions:", ("All","Gender","Religion","Race","Profession"))
detector = Detector(level,dimension)
target_sentence = st.text_input("Input the sentence you want to detect:")
def format_results(results):
formatted = ""
for result in results:
for text, pred in result.items():
formatted += f"**Text**: {text}\n\n"
formatted += "**Predictions**:\n"
for token, labels in pred.items():
formatted += f"- Token: `{token}`\n"
for label, score in labels.items():
formatted += f" - Label: `{label}`, Score: `{score}`\n"
return formatted
if st.button("Detect"):
results = detector.predict([target_sentence])
formatted_results = format_results(results)
st.markdown(f"## Detection Results: \n\n {formatted_results}")