nickmuchi commited on
Commit
a8b0182
1 Parent(s): 5f148df

Update functions.py

Browse files
Files changed (1) hide show
  1. 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, 0]]
77
- passages_emb = [['Represent the Financial statement for retrieval; Input: ',sentence,0] for sentence in corpus]
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