isana25 commited on
Commit
1fb2c6f
·
verified ·
1 Parent(s): 5a07078

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -52,8 +52,8 @@ def answer_query(query):
52
  return "❌ Please upload and process a PDF first."
53
 
54
  try:
55
- query_vec = model.encode([query])
56
- D, I = stored_index.search(np.array([query_vec]), k=3)
57
  top_chunks = [stored_chunks[i] for i in I[0]]
58
 
59
  context = "\n\n".join(top_chunks)
 
52
  return "❌ Please upload and process a PDF first."
53
 
54
  try:
55
+ query_vec = model.encode(query).reshape(1, -1)
56
+ D, I = stored_index.search(query_vec, k=3)
57
  top_chunks = [stored_chunks[i] for i in I[0]]
58
 
59
  context = "\n\n".join(top_chunks)