DeathBlade020 commited on
Commit
84aab3c
·
verified ·
1 Parent(s): d53317c

Update src/get_medical_system.py

Browse files
Files changed (1) hide show
  1. src/get_medical_system.py +5 -0
src/get_medical_system.py CHANGED
@@ -39,6 +39,11 @@ def get_vector_store():
39
  dimension = embeddings_array.shape[1]
40
  index = faiss.IndexFlatL2(dimension)
41
  index.add(embeddings_array.astype('float32')) # type: ignore
 
 
 
 
 
42
 
43
  # Create embedding function for new queries
44
  embeddings_function = HuggingFaceEmbeddings(
 
39
  dimension = embeddings_array.shape[1]
40
  index = faiss.IndexFlatL2(dimension)
41
  index.add(embeddings_array.astype('float32')) # type: ignore
42
+
43
+
44
+ import os
45
+ os.environ['SENTENCE_TRANSFORMERS_HOME'] = '/tmp'
46
+ os.makedirs('/tmp', exist_ok=True)
47
 
48
  # Create embedding function for new queries
49
  embeddings_function = HuggingFaceEmbeddings(