Spaces:
Sleeping
Sleeping
refresh
Browse files
app.py
CHANGED
@@ -8,7 +8,6 @@ import requests
|
|
8 |
from datasets import load_dataset
|
9 |
from fasthtml.common import * # noqa
|
10 |
from huggingface_hub import login, whoami
|
11 |
-
from rerankers import Reranker
|
12 |
|
13 |
# def get_images(query: str):
|
14 |
# url = "http://147.189.194.113:80/get_pages"
|
@@ -42,7 +41,7 @@ HF_REPO_ID_TXT = f"{hf_user}/zotero-answer-ai-texts"
|
|
42 |
abstract_ds = load_dataset(HF_REPO_ID_TXT, "abstracts")["train"]
|
43 |
article_ds = load_dataset(HF_REPO_ID_TXT, "articles")["train"]
|
44 |
|
45 |
-
ranker = Reranker("answerdotai/answerai-colbert-small-v1", model_type="colbert")
|
46 |
|
47 |
|
48 |
uri = "data/zotero-fts"
|
@@ -316,8 +315,9 @@ def log_query_and_results(query, results):
|
|
316 |
|
317 |
@rt("/search")
|
318 |
async def post(query: str):
|
319 |
-
results = retrieve_and_rerank(query)
|
320 |
image_results = get_images(query)
|
|
|
|
|
321 |
|
322 |
log_query_and_results(query, results)
|
323 |
|
|
|
8 |
from datasets import load_dataset
|
9 |
from fasthtml.common import * # noqa
|
10 |
from huggingface_hub import login, whoami
|
|
|
11 |
|
12 |
# def get_images(query: str):
|
13 |
# url = "http://147.189.194.113:80/get_pages"
|
|
|
41 |
abstract_ds = load_dataset(HF_REPO_ID_TXT, "abstracts")["train"]
|
42 |
article_ds = load_dataset(HF_REPO_ID_TXT, "articles")["train"]
|
43 |
|
44 |
+
# ranker = Reranker("answerdotai/answerai-colbert-small-v1", model_type="colbert")
|
45 |
|
46 |
|
47 |
uri = "data/zotero-fts"
|
|
|
315 |
|
316 |
@rt("/search")
|
317 |
async def post(query: str):
|
|
|
318 |
image_results = get_images(query)
|
319 |
+
# print(image_results)
|
320 |
+
results = retrieve_and_rerank(query)
|
321 |
|
322 |
log_query_and_results(query, results)
|
323 |
|