Files changed (1) hide show
  1. app.py +11 -9
app.py CHANGED
@@ -89,15 +89,17 @@ 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
- )[0]
99
- dense_embedding = np.array(dense_embedding_output, dtype=np.float64)
100
- dense_embedding = dense_embedding.tolist()
 
 
101
  # dense_embedding = [float(x) for x in dense_embedding_output[0]]
102
 
103
  text_embedding_instructions_choice = [
 
89
  indices = get_bm25_search_hits(corpus, sparse_scores, 50)
90
 
91
 
92
+ dense_embedding_output = instructor_model.predict(
93
+ query_embedding_instruction,
94
+ query_text,
95
+ api_name="/predict",
96
+ )
97
+
98
+ dense_embedding_arr = np.array(
99
+ literal_eval(dense_embedding_output)[0], dtype=np.float64
100
+ )
101
+ dense_embedding = dense_embedding_arr.tolist()
102
+ print(type(dense_embedding[2]))
103
  # dense_embedding = [float(x) for x in dense_embedding_output[0]]
104
 
105
  text_embedding_instructions_choice = [