Spaces:
Runtime error
Runtime error
Update generic.py
Browse files- generic.py +2 -2
generic.py
CHANGED
|
@@ -50,7 +50,7 @@ def generate_visual(text_batch, target_class):
|
|
| 50 |
tokens,
|
| 51 |
1)]
|
| 52 |
|
| 53 |
-
|
| 54 |
-
return token_importance
|
| 55 |
|
| 56 |
|
|
|
|
| 50 |
tokens,
|
| 51 |
1)]
|
| 52 |
|
| 53 |
+
visualization.visualize_text(vis_data_records)
|
| 54 |
+
return token_importance
|
| 55 |
|
| 56 |
|