Tuana commited on
Commit
cc5b5c1
1 Parent(s): 2d4dc51

Check document count

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -34,6 +34,7 @@ def pdf_to_document_store(pdf_file):
34
 
35
  def summarize(file):
36
  pdf_to_document_store(file)
 
37
  summaries = summarizer.predict(documents=document_store.get_all_documents(), generate_single_summary=True)
38
  st.write('Summary')
39
  for summary in summaries:
 
34
 
35
  def summarize(file):
36
  pdf_to_document_store(file)
37
+ st.write('Number of documents', document_store.get_document_count())
38
  summaries = summarizer.predict(documents=document_store.get_all_documents(), generate_single_summary=True)
39
  st.write('Summary')
40
  for summary in summaries: