Upload 3 files

rag_101/retriever.py

@@ -129,7 +129,7 @@ def load_embedding_model(
 
 
 def load_reranker_model(
-    reranker_model_name: str = "BAAI/bge-reranker-large", device: str = "cuda"
+    reranker_model_name: str = "BAAI/bge-reranker-large", device: str = "CPU"
 ) -> CrossEncoder:
     reranker_model = CrossEncoder(
         model_name=reranker_model_name, max_length=1024, device=device