Update app.py

app.py

@@ -363,7 +363,7 @@ def startBiasTest(test_sentences_df, gr1, gr2, att1, att2, model_name, progress=
     num_atts = 0
     for att, score in attrib_by_score.items():
         if att in attributes_g2:
-            per_attrib_bias_HTML_antistereo += att_bloombergViz(att, score, att_freqs[att])
+            per_attrib_bias_HTML_antistereo += att_bloombergViz(att, 1.0 - score, att_freqs[att])
             num_atts += 1
             if num_atts >= 8:
                 break
@@ -580,6 +580,8 @@ with gr.Blocks(theme=soft, title="Social Bias Testing in Language Models",
                 gr.Markdown("### Interpretation")
         with gr.Row():
             with gr.Column(scale=2):
+                gr.Markdown("**Legend**")
+                gr.HTML("<div style='height:20px;width:20px;background-color:#555;display:inline-block'></div>: Group 1 attribute <br><div style='height:20px;width:20px;background-color:#999;display:inline-block'></div>: Group 2 attribute") 
                 lbl_model_bias = gr.Markdown("**Model Bias** - % stereotyped choices (↑ more bias)")
                 model_bias_html = gr.HTML()
                 # model_bias_label = gr.Label(num_top_classes=1, label="% stereotyped choices (↑ more bias)",