Spaces:

Detsutut
/

clinical_negation_assert_with_explanation

Sleeping

Detsutut commited on Jan 26

Commit

9329025

•

1 Parent(s): 82abefe

Update explainer.py

Files changed (1) hide show

explainer.py CHANGED Viewed

@@ -1,6 +1,5 @@
 from transformers_interpret import SequenceClassificationExplainer
 from captum.attr import visualization as viz
-import html
 class CustomExplainer(SequenceClassificationExplainer):
@@ -90,7 +89,7 @@ class CustomExplainer(SequenceClassificationExplainer):
             attention_html.append(
                 f"<span class='word-box' style='background-color: {color};''>{word}</span><span>&nbsp;</span>")
-        attention_html = html.unescape("".join(attention_html))
         final_html = f"""
             <!DOCTYPE html>

 from transformers_interpret import SequenceClassificationExplainer
 from captum.attr import visualization as viz
 class CustomExplainer(SequenceClassificationExplainer):
             attention_html.append(
                 f"<span class='word-box' style='background-color: {color};''>{word}</span><span>&nbsp;</span>")
+        #attention_html = html.unescape("".join(attention_html))
         final_html = f"""
             <!DOCTYPE html>