Change st.markdown to st.header

app.py

@@ -4,7 +4,6 @@ from typing import Dict, List
 import pandas as pd
 import plotly.figure_factory as ff
 import plotly.graph_objects as go
-import pyperclip
 import streamlit as st
 from unidecode import unidecode
 
@@ -63,13 +62,11 @@ with description:
     analyze contracts and understand what they agree upon. 
     """
     st.write(desc)
-    st.markdown(
-        "<h1 style='text-align: center; color: black;'>Dataset statistics</h1>",
-        unsafe_allow_html=True,
-    )
+
+st.header("Dataset statistics")
 
 with dataset_statistics:
-    st.header("Number of samples in each data split")
+    st.subheader("Number of samples in each data split")
     metrics_df = pd.DataFrame.from_dict(
         {
             "Train": DATA_DICT["train"].shape[0],
@@ -96,7 +93,7 @@ with dataset_statistics:
 
 # Class distribution in each subset
 with class_distribution:
-    st.header("Class distribution in each subset")
+    st.subheader("Class distribution in each subset")
     plot_column, table_column = st.columns(2)
     with plot_column:
         hist = (