Spaces:
Sleeping
Sleeping
Runnies23
commited on
Commit
·
14aef32
1
Parent(s):
ad60ffd
Add application file
Browse files- inference_script.py +1 -1
inference_script.py
CHANGED
|
@@ -35,7 +35,7 @@ def answer_question(question , model , rerankmodel , corpus_embed , corpus_list,
|
|
| 35 |
|
| 36 |
topkindex = 15
|
| 37 |
# topk15scoresimilar_BGM3 = np.argsort(ALL_final_ans_list_ALL)[:,-topkindex:]
|
| 38 |
-
topk15scoresimilar_BGM3 = np.argsort(BGM3similarity)[
|
| 39 |
|
| 40 |
|
| 41 |
BGM3_1_retrieval = [corpus_list[i] for i in topk15scoresimilar_BGM3[0]]
|
|
|
|
| 35 |
|
| 36 |
topkindex = 15
|
| 37 |
# topk15scoresimilar_BGM3 = np.argsort(ALL_final_ans_list_ALL)[:,-topkindex:]
|
| 38 |
+
topk15scoresimilar_BGM3 = np.argsort(BGM3similarity)[:,-topkindex:]
|
| 39 |
|
| 40 |
|
| 41 |
BGM3_1_retrieval = [corpus_list[i] for i in topk15scoresimilar_BGM3[0]]
|