update

app.py

@@ -18,7 +18,11 @@ from src.prompts import wrap_prompt
 from gradio_highlightedtextbox import HighlightedTextbox
 from examples import run_example_1, run_example_2, run_example_3, run_example_4, run_example_5, run_example_6
 from functools import partial
-nltk.download('punkt')
+os.makedirs("/home/user/nltk_data", exist_ok=True)
+# Download punkt to a known path
+nltk.download("punkt", download_dir="/home/user/nltk_data")
+# Tell nltk where to find it
+nltk.data.path.append("/home/user/nltk_data")
 
 # Load original app constants
 APP_TITLE = '<div class="app-title"><span class="brand">AttnTrace: </span><span class="subtitle">Attention-based Context Traceback for Long-Context LLMs</span></div>'