Spaces:

Wills17
/

Document-RAG-System

Running

Wills17 commited on Oct 9

Commit

c381b38

verified ·

1 Parent(s): fffc3d6

Update flask_app.py

Files changed (1) hide show

flask_app.py CHANGED Viewed

@@ -85,7 +85,7 @@ def upload_file():
     # Light embedding model (fast + low memory)
     try:
-        embeds = HuggingFaceEmbeddings(model_name="sentence-transformers/paraphrase-MiniLM-L3-v2")
         vector_store = FAISS.from_documents(chunks, embeds)
         retriever = vector_store.as_retriever(search_type="similarity", search_kwargs={"k": 4})

     # Light embedding model (fast + low memory)
     try:
+        embeds = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
         vector_store = FAISS.from_documents(chunks, embeds)
         retriever = vector_store.as_retriever(search_type="similarity", search_kwargs={"k": 4})