cifkao commited on
Commit
bc6cc98
1 Parent(s): af8abd6
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -44,7 +44,13 @@ def ids_to_readable_tokens(tokenizer, ids, strip_whitespace=False):
44
  compact_layout = st.experimental_get_query_params().get("compact", ["false"]) == ["true"]
45
 
46
  if not compact_layout:
47
- st.header("Context length probing")
 
 
 
 
 
 
48
 
49
  model_name = st.selectbox("Model", ["distilgpt2", "gpt2", "EleutherAI/gpt-neo-125m"])
50
  metric_name = st.selectbox("Metric", ["KL divergence", "Cross entropy"], index=1)
 
44
  compact_layout = st.experimental_get_query_params().get("compact", ["false"]) == ["true"]
45
 
46
  if not compact_layout:
47
+ st.title("Context length probing")
48
+ st.markdown(
49
+ """[📃 Paper](https://arxiv.org/abs/2212.14815) |
50
+ [🌍 Website](https://cifkao.github.io/context-probing) |
51
+ [🧑‍💻 Code](https://cifkao.github.io/context-probing)
52
+ """
53
+ )
54
 
55
  model_name = st.selectbox("Model", ["distilgpt2", "gpt2", "EleutherAI/gpt-neo-125m"])
56
  metric_name = st.selectbox("Metric", ["KL divergence", "Cross entropy"], index=1)