Spaces:
Running
on
Zero
Running
on
Zero
Update main.py
Browse files
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]]
|