Tuana commited on
Commit
d55aa29
1 Parent(s): c94a954

trying a new summarizer model

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -17,7 +17,7 @@ def start_haystack():
17
  split_length=200,
18
  split_respect_sentence_boundary=True,
19
  )
20
- summarizer = TransformersSummarizer(model_name_or_path="google/pegasus-newsroom")
21
  return document_store, summarizer, preprocessor
22
 
23
 
@@ -44,7 +44,6 @@ document_store, summarizer, preprocessor = start_haystack()
44
  uploaded_files = st.file_uploader("Choose PDF files", accept_multiple_files=True)
45
 
46
  if uploaded_files is not None:
47
- st.write(len(uploaded_files))
48
  if st.button('Summarize Documents'):
49
  summarize(uploaded_files)
50
 
@@ -52,4 +51,4 @@ if st.button('Calculate num of docs'):
52
  st.write(document_store.get_document_count())
53
 
54
  if st.button('Clear DocumentStore'):
55
- document_store.delete_all_documents()
 
17
  split_length=200,
18
  split_respect_sentence_boundary=True,
19
  )
20
+ summarizer = TransformersSummarizer(model_name_or_path="facebook/bart-large-cnn")
21
  return document_store, summarizer, preprocessor
22
 
23
 
 
44
  uploaded_files = st.file_uploader("Choose PDF files", accept_multiple_files=True)
45
 
46
  if uploaded_files is not None:
 
47
  if st.button('Summarize Documents'):
48
  summarize(uploaded_files)
49
 
 
51
  st.write(document_store.get_document_count())
52
 
53
  if st.button('Clear DocumentStore'):
54
+ document_store.delete_documents()