racist scores

app.py

@@ -32,9 +32,9 @@ def racism_analysis(text):
             racism_analysis_pipe(sentence, return_all_scores=True)[
             0][1]["score"]
         if words[word] > 10:
-            color = "red"
+            color = "Racist"
         elif words[word] > 2:
-            color = "pink"
+            color = "Unknown"
         else:
             color = None
 
@@ -56,7 +56,7 @@ gradio_ui = gr.Interface(
         gr.outputs.Textbox(label="Label"),
         gr.outputs.Textbox(label="Racism score (0 - 100)"),
         gr.outputs.HighlightedText(
-            label="Racist heatmap", color_map={"red": "red", "pink": "pink"}),
+            label="Racist heatmap", color_map={"Racist": "red", "Unknown": "pink"}),
     ],
     examples=[
         ["Unos menas roban a una mujer"],