import streamlit as st
from utils import memory_moe_mlp, memory_mlp_layer, memory_for_attention_layer

st.title("Model Memory Usage Calculator")

x = st.slider('Select a value')
hidden_size = st.slider("The Hidden size (d_model | d)", min_value=128, step=128)

st.write(x, 'squared is', x * x)