Spaces:
Runtime error
Runtime error
Paula Leonova
commited on
Commit
•
04f7977
1
Parent(s):
3b7e91b
Update description and model loading message
Browse files
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
|
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:
|