albhu commited on
Commit
ab33443
·
verified ·
1 Parent(s): 2002500
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -91,7 +91,7 @@ def main():
91
  for uploaded_file in uploaded_files:
92
  paragraphs = search.read_pdf_pdfminer(uploaded_file) if uploaded_file.type == "application/pdf" else search.read_docx(uploaded_file)
93
  temp_df = pd.DataFrame(
94
- [(p.page_num, p.paragraph_num, p.content, search.count_tokens(p.content))
95
  for p in paragraphs],
96
  columns=["page_num", "paragraph_num", "content", "tokens"]
97
  )
@@ -131,4 +131,4 @@ def main():
131
  st.markdown("<h2>Please upload a document to proceed.</h2>", unsafe_allow_html=True)
132
 
133
  if __name__ == "__main__":
134
- main()
 
91
  for uploaded_file in uploaded_files:
92
  paragraphs = search.read_pdf_pdfminer(uploaded_file) if uploaded_file.type == "application/pdf" else search.read_docx(uploaded_file)
93
  temp_df = pd.DataFrame(
94
+ [(p.page_num, p.paragraph_num, p.content, search.count_tokens(p.content, search.tokenizer))
95
  for p in paragraphs],
96
  columns=["page_num", "paragraph_num", "content", "tokens"]
97
  )
 
131
  st.markdown("<h2>Please upload a document to proceed.</h2>", unsafe_allow_html=True)
132
 
133
  if __name__ == "__main__":
134
+ main()