UI updates

app.py

@@ -27,6 +27,9 @@ precision_options = {
     'half': 2
 }
 
+# Streamlit app
+st.title("Memory Usage Calculator for Large Language Models")
+
 # Constants from the paper, now exposed as user inputs
 layers = st.number_input("Number of Layers", value=32, step=1)
 attention_heads = st.number_input("Number of Attention Heads", value=32, step=1)
@@ -64,9 +67,6 @@ def calculate_activations(parameter_count, context_length, batch_size, vocab_siz
 
     return total_activations
 
-# Streamlit app
-st.title("Memory Usage Calculator for Large Language Models")
-
 # User inputs
 parameter_count = st.number_input("Parameter Count (in billions)", value=1, step=1) * 1e9
 context_length = st.number_input("Context Length (number of tokens)", value=512, step=1)