Spaces:
Sleeping
Sleeping
Update rag/retriever.py
Browse files- rag/retriever.py +1 -1
rag/retriever.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
from
|
| 2 |
|
| 3 |
def retrieve(query: str, top_k=8):
|
| 4 |
return search(query, top_k=top_k)
|
|
|
|
| 1 |
+
from irpr.deps import search
|
| 2 |
|
| 3 |
def retrieve(query: str, top_k=8):
|
| 4 |
return search(query, top_k=top_k)
|