Update src/streamlit_app.py

src/streamlit_app.py

@@ -575,6 +575,12 @@ def main():
             for i, (tok, prob) in enumerate(zip(top_tokens, top_probs.float().cpu().numpy()), 1):
                 st.write(f"{i}. P(**{repr(tok)}**)={prob:.3f}")
 
+    st.divider()
+    st.caption(
+        "© 2026 Toni Jianbang Liu. All rights reserved. "
+        "Cite us at https://arxiv.org/abs/2601.16407"
+    )
+
 
 if __name__ == "__main__":
     main()