Spaces:
Sleeping
Sleeping
Update flask_app.py
Browse files- 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 |
-
|
| 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":
|
| 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)
|