refresh

app.py

@@ -8,7 +8,6 @@ import requests
 from datasets import load_dataset
 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"
@@ -42,7 +41,7 @@ HF_REPO_ID_TXT = f"{hf_user}/zotero-answer-ai-texts"
 abstract_ds = load_dataset(HF_REPO_ID_TXT, "abstracts")["train"]
 article_ds = load_dataset(HF_REPO_ID_TXT, "articles")["train"]
 
-ranker = Reranker("answerdotai/answerai-colbert-small-v1", model_type="colbert")
+# ranker = Reranker("answerdotai/answerai-colbert-small-v1", model_type="colbert")
 
 
 uri = "data/zotero-fts"
@@ -316,8 +315,9 @@ def log_query_and_results(query, results):
 
 @rt("/search")
 async def post(query: str):
-    results = retrieve_and_rerank(query)
     image_results = get_images(query)
+    # print(image_results)
+    results = retrieve_and_rerank(query)
 
     log_query_and_results(query, results)