Skip to content

Commit 651b60a

Browse files
committed
fix(app): Pass missing arguments
1 parent 1486395 commit 651b60a

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

demo/app.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import pymupdf
55
import streamlit as st
66

7+
from structured_qa.config import ANSWER_PROMPT, FIND_PROMPT
78
from structured_qa.model_loaders import load_llama_cpp_model
89
from structured_qa.preprocessing import document_to_sections_dir
910
from structured_qa.workflow import find_retrieve_answer
@@ -51,6 +52,8 @@ def convert_to_sections(uploaded_file, output_dir):
5152
model=model,
5253
sections_dir=f"example_outputs/{uploaded_file.name}",
5354
question=question,
55+
find_prompt=FIND_PROMPT,
56+
answer_prompt=ANSWER_PROMPT,
5457
)
5558
st.text("Sections checked:")
5659
st.json(sections_checked)

0 commit comments

Comments
 (0)