xavierbarbier commited on
Commit
ee2bff4
·
verified ·
1 Parent(s): f6305a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,7 +48,7 @@ embeddings = HuggingFaceEmbeddings(
48
  encode_kwargs=encode_kwargs
49
  )
50
 
51
- chunk_size = 2048
52
 
53
  # creating a pdf reader object
54
 
@@ -89,7 +89,7 @@ def qa(text, question):
89
 
90
  question_embeddings = np.array([get_text_embedding(question)])
91
 
92
- D, I = index.search(question_embeddings, k=2) # distance, index
93
  retrieved_chunk = [chunks[i] for i in I.tolist()[0]]
94
 
95
  prompt = f"""
 
48
  encode_kwargs=encode_kwargs
49
  )
50
 
51
+ chunk_size = 500
52
 
53
  # creating a pdf reader object
54
 
 
89
 
90
  question_embeddings = np.array([get_text_embedding(question)])
91
 
92
+ D, I = index.search(question_embeddings, k=1) # distance, index
93
  retrieved_chunk = [chunks[i] for i in I.tolist()[0]]
94
 
95
  prompt = f"""