Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|
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 =
|
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'])
|