Update app.py

app.py

@@ -34,7 +34,7 @@ mha_flop = 2*bs*n*d*3*d
 mha_bytes = 2*bs*n*d + 2*3*d*d + 2*bs*n*3*d
 
 st.subheader("Multi-query Attention")
-c1, c2 = st.columns(columns or [2, 3])
+c1, c2 = st.columns([2, 3])
 c1.write("FLOP:")
 c2.write(str(mha_flop))
 c1.write("Bytes: ")