ShynBui commited on
Commit
4433ef7
1 Parent(s): 54c3758

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,11 +8,11 @@ import time
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]
16
  )
17
 
18
 
 
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.2, 0.8]
16
  )
17
 
18