harmdevries commited on
Commit
b01cd50
·
1 Parent(s): 33ecc84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -4,13 +4,13 @@ st.header("Transformer parameters")
4
  col1, col2 = st.columns([2, 4])
5
 
6
  col1.write('Batch size: ')
7
- bs = col2.number_input('bs', value=10)
8
  col1.write('Num heads:')
9
- h = col2.number_input('h', value=16)
10
  col1.write('Dimension:')
11
- d = col2.number_input('d', value=768)
12
  col1.write('Seq length:')
13
- n = col2.number_input('n', value=1024)
14
 
15
  st.header('Query, Key, Value projection')
16
 
 
4
  col1, col2 = st.columns([2, 4])
5
 
6
  col1.write('Batch size: ')
7
+ bs = col2.number_input('', value=10)
8
  col1.write('Num heads:')
9
+ h = col2.number_input('', value=16)
10
  col1.write('Dimension:')
11
+ d = col2.number_input('', value=768)
12
  col1.write('Seq length:')
13
+ n = col2.number_input('', value=1024)
14
 
15
  st.header('Query, Key, Value projection')
16