update

app.py

@@ -3,7 +3,14 @@ from bias_detector import Detector
 
 st.title("Multidimensional Multilevel Bias Detection")
 
-
+EXAMPLES = {
+    "Gender": "The stepfather burst into little Mikey's room with whiskey on his breath. But soon thereafter, the boy was greeted with a big surprise as many other family members plowed into his room with birthday card & presents.",
+    "Religion": "The muslim roisterer in the news was out on new year's eve making the rounds.",
+    "Race": "I had to travel to Qatar for work. It was hot, smelly, and oppressive.",
+    "Profession": "The clerk was working the cash register. She was from some Arab nation.",
+    "Input your own": ""
+    # Add more examples as needed
+}
 def format_results(results):
     formatted = ""
     for result in results:
@@ -40,7 +47,8 @@ if st.button("Load Models"):
     else:
         st.text("Failed to load models. Please check the server and/or model parameters.")
 
-target_sentence = st.text_input("Input the sentence you want to detect:")
+example_type = st.selectbox("Choose an example type:", list(EXAMPLES.keys()))
+target_sentence = st.text_input("Input the sentence you want to detect:", value=EXAMPLES[example_type])
 
 if st.button("Detect"):
     results = detector.predict([target_sentence])

bias_detector/bias_detector.py

@@ -67,13 +67,11 @@ class Detector:
         # Create the API endpoint from the model path
         self.API_URL = f"https://api-inference.huggingface.co/models/{self.model_path}"
         API_token = os.getenv("BIAS_DETECTOR_API_KEY")
+        #API_token = "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"
+
         # Add authorization token (if required)
         self.headers = {"Authorization": f"Bearer {API_token}"} # Replace `your_api_token` with your token
 
-    import time
-
-    import time
-
     def query(self, payload, max_retries=5, wait_time=5):
         retries = 0