Spaces:

WwYc
/

explain-BERT-twoclass

Runtime error

WwYc commited on Mar 6, 2024

Commit

8797a1a

verified ·

1 Parent(s): 442e1a7

Update generic.py

Files changed (1) hide show

generic.py CHANGED Viewed

@@ -49,7 +49,7 @@ def generate_visual(text_batch, target_class):
         1,
         tokens,
         1)]
-    visualization.visualize_text(vis_data_records)
-    return token_importance

         1,
         tokens,
         1)]
+    html_page = visualization.visualize_text(vis_data_records)
+    return token_importance, html_page.data