Upload retriever.py

rag_101/retriever.py

@@ -114,7 +114,7 @@ def retrieve_context(query, retriever, reranker_model):
 
 
 def load_embedding_model(
-    model_name: str = "BAAI/bge-large-en-v1.5", device: str = "cuda"
+    model_name: str = "BAAI/bge-large-en-v1.5", device: str = "CPU"
 ) -> HuggingFaceEmbeddings:
     model_kwargs = {"device": device}
     encode_kwargs = {