add header

app.py

@@ -148,7 +148,8 @@ 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
-slider_value = st.slider('Amount of text to generate', min_value=20, max_value=200, value=50, step=5)
+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'):
     context = torch.zeros((1, 1), dtype=torch.long, device=device)
     text = model.generate(context, max_new_tokens=slider_value)[0].tolist()