Wills17 commited on
Commit
90bd68b
·
verified ·
1 Parent(s): 19c5cfa

Update flask_app.py

Browse files
Files changed (1) hide show
  1. flask_app.py +3 -3
flask_app.py CHANGED
@@ -86,10 +86,10 @@ def upload_file():
86
 
87
  # Light embedding model (fast + low memory)
88
  try:
89
- # embeds = HuggingFaceEmbeddings(model_name="sentence-transformers/paraphrase-MiniLM-L3-v2")
90
- embeds = HuggingFaceEmbeddings(model_name="./models/paraphrase-MiniLM-L3-v2")
91
  vector_store = FAISS.from_documents(chunks, embeds)
92
- retriever = vector_store.as_retriever(search_type="similarity", search_kwargs={"k": 4})
93
 
94
  except Exception as e:
95
  os.unlink(path)
 
86
 
87
  # Light embedding model (fast + low memory)
88
  try:
89
+ embeds = HuggingFaceEmbeddings(model_name="sentence-transformers/paraphrase-MiniLM-L3-v2")
90
+ # embeds = HuggingFaceEmbeddings(model_name="./models/paraphrase-MiniLM-L3-v2")
91
  vector_store = FAISS.from_documents(chunks, embeds)
92
+ retriever = vector_store.as_retriever(search_type="similarity", search_kwargs={"k": 5})
93
 
94
  except Exception as e:
95
  os.unlink(path)