Mori-kamiyama commited on
Commit
33c61a3
·
1 Parent(s): 361f8fa

RAGをつけた

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def search(query):
43
  query_embedding = model.encode([query], normalize_embeddings=True)
44
 
45
  # FAISSで検索
46
- top_k = 3
47
  D, I = index.search(np.array(query_embedding), top_k)
48
 
49
  retrieved_docs = []
 
43
  query_embedding = model.encode([query], normalize_embeddings=True)
44
 
45
  # FAISSで検索
46
+ top_k = 2
47
  D, I = index.search(np.array(query_embedding), top_k)
48
 
49
  retrieved_docs = []