ShynBui commited on
Commit
038c95d
1 Parent(s): 92ec434

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ import time
8
  from transformers import AutoTokenizer, AutoModelForQuestionAnswering
9
  import torch
10
 
11
- retriever = load_the_embedding_retrieve(is_ready=False, k=3)
12
- bm25_retriever = load_the_bm25_retrieve(k=3)
13
 
14
  ensemble_retriever = EnsembleRetriever(
15
  retrievers=[bm25_retriever, retriever], weights=[0.5, 0.5]
 
8
  from transformers import AutoTokenizer, AutoModelForQuestionAnswering
9
  import torch
10
 
11
+ retriever = load_the_embedding_retrieve(is_ready=False, k=5)
12
+ bm25_retriever = load_the_bm25_retrieve(k=5)
13
 
14
  ensemble_retriever = EnsembleRetriever(
15
  retrievers=[bm25_retriever, retriever], weights=[0.5, 0.5]