Update app.py

app.py

@@ -1,9 +1,7 @@
 import streamlit as st
 
-def number_field(label, **kwargs):
-  c1, c2 = st.columns([2, 4])
+def number_field(label, c1, c2, **kwargs):
   c1.write(label)
-  
   return c2.number_input('', **kwargs)
 
 def calc_exec_time(comp_flop, mem_bytes):
@@ -30,15 +28,15 @@ def print_kernel_execution(c1, c2, comp_flop, mem_bytes):
 TFLOPS = 312e12
 GB_S = 1935e9
 
-st.header("Transformer parameters")
-col1, col2 = st.columns([2, 4])
+st.sidebar.header("Transformer parameters")
+col1, col2 = st.sidebar.columns([2, 4])
 
-bs = number_field('Batch size', value=10)
-h = number_field('Num heads', value=16)
-d = number_field('Dimension', value=768)
-n_start = number_field('Start seq', value=1)
-n = number_field('End seq', value=1024)
-l = number_field('Num layers', value=24)
+bs = number_field('Batch size', col1, col2, value=10)
+h = number_field('Num heads', col1, col2, value=16)
+d = number_field('Dimension', col1, col2, value=768)
+n_start = number_field('Start seq', col1, col2, value=1)
+n = number_field('End seq', col1, col2, value=1024)
+l = number_field('Num layers', col1, col2, value=24)
 
 st.header('Attention layer')