Update app.py

app.py

@@ -178,6 +178,7 @@ def main():
                     explainer = TextExplainer(sampler=MaskingTextSampler())
                     explainer.fit(st.session_state["text"], model.predict_proba)
                     html = eli5.format_as_html(explainer.explain_prediction(target_names=["Human", "AI"]))
+                    st.markdown('<span style="color:green"><strong>Green:</strong> Contributes to decision | </span><span style="color:red"><strong>Red:</strong> Opposite</span>', unsafe_allow_html=True)
             else:
                 with st.spinner('Wait for it 💭... BERT-based model explanations take around 4-10 minutes. In case you want to abort, refresh the page.'):
                 # TORCH EXPLAINER PRED FUNC (USES logits)