Update app.py

app.py

@@ -96,7 +96,7 @@ st.write("We also need to take into the softmax layer and layer norm")
 st.caption("Softmax")
 softmax_bytes = 2*bs*h*n + 2*bs*h*n
 c1, c2 = st.columns([2, 3])
-softmax_time = print_kernel_execution(c1, c2, 0, softmax_bytes))
+softmax_time = print_kernel_execution(c1, c2, 0, softmax_bytes)
 
 st.caption("Layer norm")