kristada673 commited on
Commit
4af7d90
·
1 Parent(s): 8f0f959

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,11 +1,11 @@
1
  import os, gradio
2
  from langchain.document_loaders import UnstructuredPDFLoader
3
  from langchain.indexes import VectorstoreIndexCreator
 
4
 
5
  os.environ['OPENAI_API_KEY'] = os.getenv("OPENAI_API_KEY")
6
 
7
- text_folder = '.'
8
- loaders = [UnstructuredPDFLoader(os.path.join(text_folder, fn)) for fn in os.listdir(text_folder)]
9
 
10
  # Create the index, if it does not exist, and save it
11
  if not os.path.isfile('VectorStoreIndex/chroma-embeddings.parquet'):
 
1
  import os, gradio
2
  from langchain.document_loaders import UnstructuredPDFLoader
3
  from langchain.indexes import VectorstoreIndexCreator
4
+ import glob
5
 
6
  os.environ['OPENAI_API_KEY'] = os.getenv("OPENAI_API_KEY")
7
 
8
+ loaders = [UnstructuredPDFLoader(glob.glob("*.pdf"))]
 
9
 
10
  # Create the index, if it does not exist, and save it
11
  if not os.path.isfile('VectorStoreIndex/chroma-embeddings.parquet'):