diff --git a/atman-magma/example_document_qa_sentence_level_explain.py b/atman-magma/example_document_qa_sentence_level_explain.py index 9ce4183..70c1961 100755 --- a/atman-magma/example_document_qa_sentence_level_explain.py +++ b/atman-magma/example_document_qa_sentence_level_explain.py @@ -49,7 +49,7 @@ document_explainer = DocumentQAExplainer( model, - explanation_delimiter = '\n', ## for now max len = 1 token (todo: add support for longer delimiters) + explanation_delimiters = ['\n'], ## for now max len = 1 token (todo: add support for longer delimiters) device = 'cuda:0', suppression_factor = 0.0, ## suppression_factor = 0.0 -> completely erases one chunk at a time document = DOCUMENT @@ -76,3 +76,4 @@ ## save as json or print postprocessed_output = document_explainer.postprocess(output) postprocessed_output.save_as(filename = 'explanation.json') +print('explanation.json')