SecureLLMSys commited on
Commit
48f2e3c
·
1 Parent(s): ee19553
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -18,6 +18,7 @@ from src.prompts import wrap_prompt
18
  from gradio_highlightedtextbox import HighlightedTextbox
19
  from examples import run_example_1, run_example_2, run_example_3, run_example_4, run_example_5, run_example_6
20
  from functools import partial
 
21
 
22
  # Load original app constants
23
  APP_TITLE = '<div class="app-title"><span class="brand">AttnTrace: </span><span class="subtitle">Attention-based Context Traceback for Long-Context LLMs</span></div>'
 
18
  from gradio_highlightedtextbox import HighlightedTextbox
19
  from examples import run_example_1, run_example_2, run_example_3, run_example_4, run_example_5, run_example_6
20
  from functools import partial
21
+ nltk.download('punkt')
22
 
23
  # Load original app constants
24
  APP_TITLE = '<div class="app-title"><span class="brand">AttnTrace: </span><span class="subtitle">Attention-based Context Traceback for Long-Context LLMs</span></div>'