Update app.py

app.py

@@ -192,6 +192,7 @@ def main():
                     explainer = shap.Explainer(f, tokenizer)
                     shap_values = explainer([st.session_state["text"]], fixed_context=1)
                     html = shap.plots.text(shap_values, display=False)
+                    st.markdown('<span style="color:blue"><strong>Blue:</strong> Contributes to "human" | </span><span style="color:red"><strong>Red:</strong> Contributes to "AI"</span>', unsafe_allow_html=True)
             # Render HTML
             st.components.v1.html(html, height=500, scrolling = True)
         else: