nickmuchi commited on
Commit
e7363fe
1 Parent(s): d5afb66

Update functions.py

Browse files
Files changed (1) hide show
  1. functions.py +2 -2
functions.py CHANGED
@@ -63,8 +63,8 @@ def embed_text(query,corpus,embedding_model):
63
  passages_emb = ['passage: ' + sentence for sentence in corpus]
64
 
65
  elif embedding_model == 'hkunlp/instructor-base':
66
- search_input = [['Represent the Financial question for retrieving supporting documents; Input: ', query, 0]]
67
- passages_emb = [['Represent the Financial document for retrieval; Input: ',sentence,0] for sentence in corpus]
68
 
69
  else:
70
  search_input = query
 
63
  passages_emb = ['passage: ' + sentence for sentence in corpus]
64
 
65
  elif embedding_model == 'hkunlp/instructor-base':
66
+ search_input = [['Represent the Financial question; Input: ', query, 0]]
67
+ passages_emb = [['Represent the Financial statement for retrieval; Input: ',sentence,0] for sentence in corpus]
68
 
69
  else:
70
  search_input = query