razent commited on
Commit
f769ccc
·
1 Parent(s): a424978

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,9 +19,9 @@ def predict(input_ids, attention_mask):
19
  return res
20
 
21
  if __name__ == '__main__':
22
- st.title("ViT5 News Abstractive Summarization (Vietnamese)")
23
  with st.container():
24
- txt = st.text_area('Enter long documment...', ' ')
25
  inp_ids, attn_mask = preprocess(txt)
26
  st.write('Summary:', predict(inp_ids, attn_mask))
27
 
 
19
  return res
20
 
21
  if __name__ == '__main__':
22
+ st.title("ViT5 News Abstractive Summarization")
23
  with st.container():
24
+ txt = st.text_area('Enter a long Vietnamese document...', ' ')
25
  inp_ids, attn_mask = preprocess(txt)
26
  st.write('Summary:', predict(inp_ids, attn_mask))
27