broskicodes commited on
Commit
1770e8b
·
1 Parent(s): 7f7b222

add missing colon

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -149,7 +149,7 @@ model = torch.load('complete-model.pt')
149
 
150
  # inference
151
  slider_value = st.slider('Amount of text to generate', min_value=100, max_value=2000, value=500, step=5)
152
- if st.button('Generat text')
153
  context = torch.zeros((1, 1), dtype=torch.long, device='cuda')
154
  text = model.generate(context, max_new_tokens=slider_value)[0].tolist()
155
  st.json(decode(text))
 
149
 
150
  # inference
151
  slider_value = st.slider('Amount of text to generate', min_value=100, max_value=2000, value=500, step=5)
152
+ if st.button('Generat text'):
153
  context = torch.zeros((1, 1), dtype=torch.long, device='cuda')
154
  text = model.generate(context, max_new_tokens=slider_value)[0].tolist()
155
  st.json(decode(text))