Spaces:

Sriramsr3
/

InsureRAG

Sleeping

Sriramsr3 commited on 6 days ago

Commit

ea628b4

verified ·

1 Parent(s): 5b66090

Update rag_chain.py

Files changed (1) hide show

rag_chain.py CHANGED Viewed

@@ -133,7 +133,7 @@ def query_preprocessor(query: str) -> str:
 # === Load and process PDF ===
 def build_rag_chain(pdf_path: str, rebuild_index=False):
-    embeddings = SentenceTransformerEmbeddings(model_name="nomic-ai/nomic-embed-text-v1.5")
     final_chunks = []  # ✅ ensure it's always defined
     if not rebuild_index and os.path.exists("/tmp/faiss_index"):

 # === Load and process PDF ===
 def build_rag_chain(pdf_path: str, rebuild_index=False):
+    embeddings = SentenceTransformerEmbeddings(model_name="intfloat/e5-small-v2")
     final_chunks = []  # ✅ ensure it's always defined
     if not rebuild_index and os.path.exists("/tmp/faiss_index"):