Spaces:
Runtime error
Runtime error
fix error
Browse files- .gitignore +1 -0
- src/rag_engine.py +4 -2
.gitignore
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
__pycache__/
|
src/rag_engine.py
CHANGED
|
@@ -1,12 +1,14 @@
|
|
| 1 |
import json
|
| 2 |
import os
|
| 3 |
import faiss
|
| 4 |
-
import numpy as np
|
| 5 |
from sentence_transformers import SentenceTransformer
|
| 6 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
| 7 |
|
| 8 |
# Load embedding model
|
| 9 |
-
|
|
|
|
|
|
|
|
|
|
| 10 |
|
| 11 |
print(f"DebugLogs___ Current position: {os.getcwd()}")
|
| 12 |
# Load your JSON KB
|
|
|
|
| 1 |
import json
|
| 2 |
import os
|
| 3 |
import faiss
|
|
|
|
| 4 |
from sentence_transformers import SentenceTransformer
|
| 5 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
| 6 |
|
| 7 |
# Load embedding model
|
| 8 |
+
os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers"
|
| 9 |
+
os.environ["HF_HOME"] = "/tmp/huggingface"
|
| 10 |
+
|
| 11 |
+
embed_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", cache_folder="/tmp/sentence_transformers")
|
| 12 |
|
| 13 |
print(f"DebugLogs___ Current position: {os.getcwd()}")
|
| 14 |
# Load your JSON KB
|