Spaces:
Sleeping
Sleeping
Commit
·
9c46b79
1
Parent(s):
1e93f98
Minor fix
Browse files
app.py
CHANGED
|
@@ -26,7 +26,9 @@ prompt = PromptTemplate(template = prompt_template, input_variables = ["context"
|
|
| 26 |
chain = prompt | gemini
|
| 27 |
|
| 28 |
def inference(pdf_path, chunk_size, chunk_overlap):
|
| 29 |
-
raw_documents =
|
|
|
|
|
|
|
| 30 |
text_splitter = CharacterTextSplitter(chunk_size=chunk_size, chunk_overlap=chunk_overlap)
|
| 31 |
documents = text_splitter.split_documents(raw_documents)
|
| 32 |
|
|
@@ -57,7 +59,7 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
|
|
| 57 |
gr.Markdown(f"# {title}\n{description}")
|
| 58 |
with gr.Row():
|
| 59 |
with gr.Column():
|
| 60 |
-
pdf = gr.File(label="
|
| 61 |
chunk_size = gr.Slider(0, 2000, 1000, 100, label="Size of Chunk")
|
| 62 |
chunk_overlap = gr.Slider(0, 1000, 100, 100, label="Size of Chunk Overlap")
|
| 63 |
with gr.Row():
|
|
|
|
| 26 |
chain = prompt | gemini
|
| 27 |
|
| 28 |
def inference(pdf_path, chunk_size, chunk_overlap):
|
| 29 |
+
raw_documents = []
|
| 30 |
+
for path in pdf_path:
|
| 31 |
+
raw_documents.extend(PyPDFLoader(path).load())
|
| 32 |
text_splitter = CharacterTextSplitter(chunk_size=chunk_size, chunk_overlap=chunk_overlap)
|
| 33 |
documents = text_splitter.split_documents(raw_documents)
|
| 34 |
|
|
|
|
| 59 |
gr.Markdown(f"# {title}\n{description}")
|
| 60 |
with gr.Row():
|
| 61 |
with gr.Column():
|
| 62 |
+
pdf = gr.File(label="Input PDFs", file_count="multiple", file_types=[".pdf"])
|
| 63 |
chunk_size = gr.Slider(0, 2000, 1000, 100, label="Size of Chunk")
|
| 64 |
chunk_overlap = gr.Slider(0, 1000, 100, 100, label="Size of Chunk Overlap")
|
| 65 |
with gr.Row():
|