Update functions.py
Browse files- functions.py +2 -2
functions.py
CHANGED
|
@@ -73,8 +73,8 @@ def embed_text(query,corpus,embedding_model):
|
|
| 73 |
passages_emb = ['passage: ' + sentence for sentence in corpus]
|
| 74 |
|
| 75 |
elif embedding_model == 'hkunlp/instructor-base':
|
| 76 |
-
search_input = [['Represent the Financial question; Input: ', query
|
| 77 |
-
passages_emb = [['Represent the Financial statement for retrieval; Input: ',sentence
|
| 78 |
|
| 79 |
else:
|
| 80 |
search_input = query
|
|
|
|
| 73 |
passages_emb = ['passage: ' + sentence for sentence in corpus]
|
| 74 |
|
| 75 |
elif embedding_model == 'hkunlp/instructor-base':
|
| 76 |
+
search_input = [['Represent the Financial question; Input: ', query]]
|
| 77 |
+
passages_emb = [['Represent the Financial statement for retrieval; Input: ',sentence] for sentence in corpus]
|
| 78 |
|
| 79 |
else:
|
| 80 |
search_input = query
|