AmirFARES commited on
Commit
8473d47
·
1 Parent(s): 73fa029

fix error

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. 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
- embed_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
 
 
 
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