nickmuchi commited on
Commit
50e39ef
1 Parent(s): 64eeae3

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 for retrieving supporting documents: ', query]]
77
- passages_emb = [['Represent the Financial document for retrieval: ',sentence] 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 for retrieving supporting paragraphs: ', query]]
77
+ passages_emb = [['Represent the Financial paragraph for retrieval: ',sentence] for sentence in corpus]
78
 
79
  else:
80
  search_input = query