Adhin commited on
Commit
2f45137
·
1 Parent(s): 8048e47

add global var

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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)