Fix code link, add label

app.py

@@ -48,7 +48,7 @@ if not compact_layout:
     st.markdown(
         """[📃 Paper](https://arxiv.org/abs/2212.14815) |
         [🌍 Website](https://cifkao.github.io/context-probing) |
-        [🧑‍💻 Code](https://cifkao.github.io/context-probing)
+        [🧑‍💻 Code](https://github.com/cifkao/context-probing)
         """
     )
 
@@ -150,4 +150,5 @@ scores = run_context_length_probing(
 )
 tokens = ids_to_readable_tokens(tokenizer, input_ids)
 
+st.markdown('<label style="font-size: 14px;">Output</label>', unsafe_allow_html=True)
 highlighted_text_component(tokens=tokens, scores=scores.tolist())