tjtanaa commited on
Commit
61a8b51
1 Parent(s): 07a6749

add jinja2 template download

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -184,6 +184,13 @@ if st.session_state['tokenizer_json'] is not None:
184
  mime='application/json',
185
  use_container_width=True
186
  )
 
 
 
 
 
 
 
187
  if push_to_hub_button:
188
  with open(f"./tmp/{st.session_state['uuid']}/tmp_chat_template.json", "w") as fp:
189
  fp.write(st.session_state['successful_template'])
 
184
  mime='application/json',
185
  use_container_width=True
186
  )
187
+ st.download_button(
188
+ label="Download chat_template.jinja2",
189
+ data=st.session_state['successful_template'],
190
+ file_name='chat_template.jinja2',
191
+ mime='text/plain',
192
+ use_container_width=True
193
+ )
194
  if push_to_hub_button:
195
  with open(f"./tmp/{st.session_state['uuid']}/tmp_chat_template.json", "w") as fp:
196
  fp.write(st.session_state['successful_template'])