Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -48,7 +48,7 @@ embeddings = HuggingFaceEmbeddings(
|
|
48 |
encode_kwargs=encode_kwargs
|
49 |
)
|
50 |
|
51 |
-
chunk_size =
|
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=
|
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"""
|