Spaces:
Paused
Paused
add global var
Browse files
app.py
CHANGED
|
@@ -59,6 +59,8 @@ def rag_with_reranking(query : str):
|
|
| 59 |
@app.get("/search")
|
| 60 |
def search_text(query):
|
| 61 |
"""Searches for similar texts."""
|
|
|
|
|
|
|
| 62 |
print(f"Searching for: {query}")
|
| 63 |
if not reranker_model:
|
| 64 |
reranker_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-v2-m3")
|
|
@@ -68,7 +70,7 @@ def search_text(query):
|
|
| 68 |
print("embedding model loaded")
|
| 69 |
|
| 70 |
results = rag_with_reranking(query)
|
| 71 |
-
|
| 72 |
lst = []
|
| 73 |
for doc in results:
|
| 74 |
lst.append(doc.page_content)
|
|
|
|
| 59 |
@app.get("/search")
|
| 60 |
def search_text(query):
|
| 61 |
"""Searches for similar texts."""
|
| 62 |
+
global reranker_model, embedding
|
| 63 |
+
|
| 64 |
print(f"Searching for: {query}")
|
| 65 |
if not reranker_model:
|
| 66 |
reranker_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-v2-m3")
|
|
|
|
| 70 |
print("embedding model loaded")
|
| 71 |
|
| 72 |
results = rag_with_reranking(query)
|
| 73 |
+
|
| 74 |
lst = []
|
| 75 |
for doc in results:
|
| 76 |
lst.append(doc.page_content)
|