ccm commited on
Commit
f23e0a7
·
verified ·
1 Parent(s): 55e25df

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +0 -1
main.py CHANGED
@@ -71,7 +71,6 @@ def preprocess(query: str, k: int) -> tuple[str, str]:
71
  tuple[str, str]: A tuple containing the prompt and references
72
  """
73
  encoded_query = numpy.expand_dims(model.encode(query), axis=0)
74
- print(query, encoded_query)
75
  faiss.normalize_L2(encoded_query)
76
  D, I = index.search(encoded_query, k)
77
  top_five = data.loc[I[0]]
 
71
  tuple[str, str]: A tuple containing the prompt and references
72
  """
73
  encoded_query = numpy.expand_dims(model.encode(query), axis=0)
 
74
  faiss.normalize_L2(encoded_query)
75
  D, I = index.search(encoded_query, k)
76
  top_five = data.loc[I[0]]