Lucas commited on
Commit
0a6abdb
·
1 Parent(s): 13b9680

Fixing directory in which PDF should be saved.

Browse files
Files changed (1) hide show
  1. main.py +2 -3
main.py CHANGED
@@ -7,7 +7,6 @@ from langchain.vectorstores import Chroma
7
  from langchain.document_loaders import PyPDFLoader
8
  import panel as pn
9
  import PyPDF2
10
- import tempfile
11
 
12
  pn.extension(notifications=True)
13
  pn.extension('texteditor', template="bootstrap", sizing_mode='stretch_width')
@@ -92,11 +91,11 @@ def qa_result(_):
92
 
93
  # save pdf file to a temp file
94
  if file_input.value is not None:
95
- file_input.save("temp.pdf")
96
 
97
  prompt_text = prompt.value
98
  if prompt_text:
99
- result = qa(file="temp.pdf", query=prompt_text, chain_type=select_chain_type.value,
100
  k=select_k.value)
101
  if result.get('error') is None:
102
  convos.extend([
 
7
  from langchain.document_loaders import PyPDFLoader
8
  import panel as pn
9
  import PyPDF2
 
10
 
11
  pn.extension(notifications=True)
12
  pn.extension('texteditor', template="bootstrap", sizing_mode='stretch_width')
 
91
 
92
  # save pdf file to a temp file
93
  if file_input.value is not None:
94
+ file_input.save("/.cache/temp.pdf")
95
 
96
  prompt_text = prompt.value
97
  if prompt_text:
98
+ result = qa(file="/.cache/temp.pdf", query=prompt_text, chain_type=select_chain_type.value,
99
  k=select_k.value)
100
  if result.get('error') is None:
101
  convos.extend([