awinml commited on
Commit
22f793f
1 Parent(s): 1e8d423

Upload 7 files

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -89,15 +89,15 @@ 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_output = literal_eval(
93
  instructor_model.predict(
94
  query_embedding_instruction,
95
  query_text,
96
  api_name="/predict",
97
  )
98
- )
99
 
100
- dense_embedding = [float(x) for x in dense_embedding_output[0]]
101
 
102
  text_embedding_instructions_choice = [
103
  "Represent the financial statement for retrieval:",
 
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,
96
  api_name="/predict",
97
  )
98
+ ).tolist()
99
 
100
+ #dense_embedding = [float(x) for x in dense_embedding_output[0]]
101
 
102
  text_embedding_instructions_choice = [
103
  "Represent the financial statement for retrieval:",