Nopphakorn commited on
Commit
6c03bb7
·
1 Parent(s): 7f42d9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -83,7 +83,7 @@ def generate_title():
83
  decoded_outputs = tokenizer.batch_decode(outputs, skip_special_tokens=True)
84
  predicted_titles = decoded_output
85
 
86
- st.session_state.titles = predicted_titles
87
 
88
  # generate title button
89
  st_generate_button = st.button('Generate headlines', on_click=generate_title)
 
83
  decoded_outputs = tokenizer.batch_decode(outputs, skip_special_tokens=True)
84
  predicted_titles = decoded_output
85
 
86
+ st.session_state.titles = decoded_outputs
87
 
88
  # generate title button
89
  st_generate_button = st.button('Generate headlines', on_click=generate_title)