Soumen commited on
Commit
ef54be3
1 Parent(s): 1e17e2b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -198,5 +198,7 @@ def main():
198
  summary_ids = model.generate(inputs, max_length=150, min_length=80, length_penalty=5., num_beams=2)
199
  summary = tokenizer.decode(summary_ids[0])
200
  st.success(summary)
 
 
201
  if __name__ == '__main__':
202
  main()
 
198
  summary_ids = model.generate(inputs, max_length=150, min_length=80, length_penalty=5., num_beams=2)
199
  summary = tokenizer.decode(summary_ids[0])
200
  st.success(summary)
201
+ if st.button("refresh"):
202
+ st.experimental_rerun()
203
  if __name__ == '__main__':
204
  main()