add color map

app.py

@@ -33,7 +33,7 @@ def racism_analysis(text):
             0][1]["score"]
         if words[word] > 10:
             color = "red"
-        elif words[word] > 3:
+        elif words[word] > 2:
             color = "pink"
         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"),
+            label="Racist heatmap", colors_map={"red": "red", "pink": "pink"}),
     ],
     examples=[
         ["Unos menas roban a una mujer"],