ishmeet-yo commited on
Commit
9eb8742
·
verified ·
1 Parent(s): 1500e64

Update app/rag.py

Browse files
Files changed (1) hide show
  1. app/rag.py +2 -2
app/rag.py CHANGED
@@ -9,8 +9,8 @@ from sklearn.preprocessing import normalize
9
 
10
  CACHE_DIR = "app/cache"
11
  DATA_DIR = "app/data"
12
- # MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
13
- MODEL_NAME = SentenceTransformer("app/models/all-MiniLM-L6-v2")
14
  print("Model loaded from local path")
15
  CHUNK_SIZE = 500
16
  CHUNK_OVERLAP = 100
 
9
 
10
  CACHE_DIR = "app/cache"
11
  DATA_DIR = "app/data"
12
+ MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
13
+ # MODEL_NAME = SentenceTransformer("app/models/all-MiniLM-L6-v2")
14
  print("Model loaded from local path")
15
  CHUNK_SIZE = 500
16
  CHUNK_OVERLAP = 100