fix

app.py

@@ -88,6 +88,8 @@ def retrieve_and_rerank(query, k=5):
         .to_list()
     )
 
+    print(f"Retrieved {len(retrieved)} documents")
+
     # re-rank
     docs = [f"{item['title']} {item['abstract']}" for item in retrieved]
     results = ranker.rank(query=query, docs=docs)

requirements.txt

@@ -11,5 +11,6 @@ datasets
 tqdm
 tantivy==0.22.0
 lancedb
-rerankers
-transformers
+rerankers==0.5.3
+transformers
+torch