Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -89,7 +89,7 @@ sparse_scores = np.argsort(bm25.get_scores(tokenized_query), axis=0)[::-1]
89
  indices = get_bm25_search_hits(corpus, sparse_scores, 50)
90
 
91
 
92
- dense_embedding = literal_eval(
93
  instructor_model.predict(
94
  query_embedding_instruction,
95
  query_text,
@@ -97,6 +97,7 @@ dense_embedding = literal_eval(
97
  )
98
  )
99
 
 
100
 
101
  text_embedding_instructions_choice = [
102
  "Represent the financial statement for retrieval:",
 
89
  indices = get_bm25_search_hits(corpus, sparse_scores, 50)
90
 
91
 
92
+ dense_embedding_output = literal_eval(
93
  instructor_model.predict(
94
  query_embedding_instruction,
95
  query_text,
 
97
  )
98
  )
99
 
100
+ dense_embedding = [int(x) for x in dense_embedding_output]
101
 
102
  text_embedding_instructions_choice = [
103
  "Represent the financial statement for retrieval:",