fracapuano commited on
Commit
ef45488
1 Parent(s): 51a7497

fix: minor debug for method renaming

Browse files
Files changed (1) hide show
  1. summarization/summarization.py +2 -2
summarization/summarization.py CHANGED
@@ -1,6 +1,6 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
- from qa.qa import file_to_doc
4
  from transformers import AutoTokenizer
5
  from typing import Text, Union
6
 
@@ -59,7 +59,7 @@ def summarization_main():
59
  )
60
  if uploaded_file is not None:
61
  # parse the file using custom parsers and build a concatenation for the summarizer
62
- text = " ".join(file_to_doc(uploaded_file))
63
  # toggle text is given greenlight
64
  text_is_given = not text_is_given
65
 
 
1
  import streamlit as st
2
  from transformers import pipeline
3
+ from qa.qa import parse_file
4
  from transformers import AutoTokenizer
5
  from typing import Text, Union
6
 
 
59
  )
60
  if uploaded_file is not None:
61
  # parse the file using custom parsers and build a concatenation for the summarizer
62
+ text = " ".join(parse_file(uploaded_file))
63
  # toggle text is given greenlight
64
  text_is_given = not text_is_given
65