add header

app.py

@@ -148,9 +148,10 @@ decode = lambda l: ''.join([itos[i] for i in l]) # decoder: take a list of integ
 model = torch.load('complete-model.pt', map_location=device)
 
 # inference
+st.markdown('## This is a simple lm for generating text in Skakespeareian style')
 st.markdown('### Generation will be slow. Please be patient :)')
 slider_value = st.slider('Amount of text to generate', min_value=100, max_value=2000, value=200, step=5)
-if st.button('Generat text'):
+if st.button('Generate text'):
     context = torch.zeros((1, 1), dtype=torch.long, device=device)
     text = model.generate(context, max_new_tokens=slider_value)[0].tolist()
     st.text(decode(text))