yashasgupta commited on
Commit
87901a4
·
verified ·
1 Parent(s): 8ae898a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -65,7 +65,7 @@ if uploaded_file is not None:
65
  #pdf_loader = PDFMinerLoader(pdf_file)
66
  #dat_nik = pdf_loader.load()
67
  text_splitter = NLTKTextSplitter(chunk_size = 500,chunk_overlap = 100)
68
- chunks = test_splitter.split_documents([text])
69
 
70
  embedding_model = GoogleGenerativeAIEmbeddings(model="models/embedding-001")
71
 
@@ -73,7 +73,7 @@ if uploaded_file is not None:
73
 
74
  db.persist()
75
 
76
- db_connection = Chroma(persist_directory="./chroma_db_", embedding_function=embedding_model)
77
 
78
  retriever = db_connection.as_retriever(search_kwargs={"k": 5})
79
 
 
65
  #pdf_loader = PDFMinerLoader(pdf_file)
66
  #dat_nik = pdf_loader.load()
67
  text_splitter = NLTKTextSplitter(chunk_size = 500,chunk_overlap = 100)
68
+ chunks = text_splitter.split_documents([text])
69
 
70
  embedding_model = GoogleGenerativeAIEmbeddings(model="models/embedding-001")
71
 
 
73
 
74
  db.persist()
75
 
76
+ db_connection = Chroma(persist_directory="./chroma_db_1", embedding_function=embedding_model)
77
 
78
  retriever = db_connection.as_retriever(search_kwargs={"k": 5})
79