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
|