Paula Leonova commited on
Commit
04f7977
1 Parent(s): 3b7e91b

Update description and model loading message

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,7 +19,7 @@ ex_long_text = example_long_text_load()
19
 
20
  # if __name__ == '__main__':
21
  st.markdown("### Long Text Summarization & Multi-Label Classification")
22
- st.write("This app summarizes and then classifies your long text with multiple labels.")
23
  st.write("__Inputs__: User enters their own custom text and labels.")
24
  st.write("__Outputs__: A summary of the text, likelihood percentages for each label and a downloadable csv of the results. \
25
  Includes additional options to generate a list of keywords and/or evaluate results against a list of ground truth labels, if available.")
@@ -74,7 +74,7 @@ with st.spinner('Loading pretrained models...'):
74
  kw_model = md.load_keyword_model()
75
  k_time = round(time.time() - start,4)
76
 
77
- st.success(f'Time taken to load KeyBERT model: {k_time}s & BART summarizer mnli model: {s_time}s & BART classifier mnli model: {c_time}s')
78
 
79
 
80
  if submit_button or example_button:
19
 
20
  # if __name__ == '__main__':
21
  st.markdown("### Long Text Summarization & Multi-Label Classification")
22
+ st.write("This app summarizes and then classifies your long text with multiple labels using [BART Large MNLI](https://huggingface.co/facebook/bart-large-mnli). The keywords are generated using [KeyBERT](https://github.com/MaartenGr/KeyBERT).")
23
  st.write("__Inputs__: User enters their own custom text and labels.")
24
  st.write("__Outputs__: A summary of the text, likelihood percentages for each label and a downloadable csv of the results. \
25
  Includes additional options to generate a list of keywords and/or evaluate results against a list of ground truth labels, if available.")
74
  kw_model = md.load_keyword_model()
75
  k_time = round(time.time() - start,4)
76
 
77
+ st.success(f'Time taken to load various models: {k_time}s for KeyBERT model & {s_time}s for BART summarizer mnli model & {c_time}s for BART classifier mnli model.')
78
 
79
 
80
  if submit_button or example_button: