Sasidhar commited on
Commit
d965516
·
1 Parent(s): 3ee5fd7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -105,7 +105,7 @@ elif selected_menu == "Extract Text":
105
  elif selected_menu == "Summarize Document":
106
  text = get_text_from_ocr_engine()
107
  with st.spinner("Summarizing Document..."):
108
- summary_text = pipeline_summarization(review, max_length=130, min_length=10, do_sample=False)
109
  # Show output
110
  st.write(summary_text[0]['summary_text'])
111
 
@@ -120,6 +120,6 @@ elif selected_menu == "Get Answers":
120
  question_text = st.text_input("Type your question")
121
  context = get_text_from_ocr_engine()
122
  if question_text:
123
- result = question_answerer(question=question_text, context=context)
124
  st.subheader('Answer')
125
  st.text(result['answer'])
 
105
  elif selected_menu == "Summarize Document":
106
  text = get_text_from_ocr_engine()
107
  with st.spinner("Summarizing Document..."):
108
+ summary_text = pipeline_summarization(text, max_length=130, min_length=10, do_sample=False)
109
  # Show output
110
  st.write(summary_text[0]['summary_text'])
111
 
 
120
  question_text = st.text_input("Type your question")
121
  context = get_text_from_ocr_engine()
122
  if question_text:
123
+ result = pipeline_qa(question=question_text, context=context)
124
  st.subheader('Answer')
125
  st.text(result['answer'])