Enable compact layout

app.py

@@ -41,7 +41,10 @@ def ids_to_readable_tokens(tokenizer, ids, strip_whitespace=False):
             result.append("")
     return result
 
-st.header("Context length probing")
+compact_layout = st.experimental_get_query_params().get("compact", ["false"]) == ["true"]
+
+if not compact_layout:
+    st.header("Context length probing")
 
 model_name = st.selectbox("Model", ["distilgpt2", "gpt2", "EleutherAI/gpt-neo-125m"])
 metric_name = st.selectbox("Metric", ["KL divergence", "Cross entropy"], index=1)