change backend

app.py

@@ -10,14 +10,29 @@ from fasthtml.common import *  # noqa
 from huggingface_hub import login, whoami
 from rerankers import Reranker
 
+# def get_images(query: str):
+#     url = "http://147.189.194.113:80/get_pages"
+
+#     response = requests.get(url, params={"query": query})
+#     return response.json()
+
 
 def get_images(query: str):
-    url = "http://147.189.194.113:80/get_pages"
+    url = "http://47.47.180.31:80/get_pages"
 
     response = requests.get(url, params={"query": query})
     return response.json()
 
 
+def rerank_api(query, docs):
+    url = "http://47.47.180.31:80/rerank"
+
+    params = {"query": query, "docs": docs}
+
+    response = requests.get(url, params=params)
+    return response.json()
+
+
 dotenv.load_dotenv()
 login(token=os.environ.get("HF_TOKEN"))
 
@@ -101,11 +116,12 @@ def retrieve_and_rerank(query, k=3):
 
     # re-rank
     docs = [f"{item['title']} {item['abstract']}" for item in retrieved]
-    results = ranker.rank(query=query, docs=docs)
+    # results = ranker.rank(query=query, docs=docs)
+    ranked_doc_ids = rerank_api(query, docs)["ranked_doc_ids"][:k]
 
-    ranked_doc_ids = []
-    for result in results[:k]:
-        ranked_doc_ids.append(result.doc_id)
+    # ranked_doc_ids = []
+    # for result in results[:k]:
+    #     ranked_doc_ids.append(result.doc_id)
 
     final_results = [retrieved[idx] for idx in ranked_doc_ids]
     final_results = _format_results(final_results)

requirements.txt

@@ -13,4 +13,5 @@ tantivy==0.22.0
 lancedb
 rerankers==0.5.3
 transformers
-torch
+torch
+pillow