aymanemalih commited on
Commit
64c3b49
·
1 Parent(s): 4a479b2

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +8 -2
main.py CHANGED
@@ -5,9 +5,15 @@ import openai
5
  import qdrant_client
6
 
7
 
8
- from sentence_transformers import SentenceTransformer
9
  #model = SentenceTransformer('paraphrase-multilingual-MiniLM-L12-v2') # good so far
10
- model = SentenceTransformer('models/sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2') # good so far
 
 
 
 
 
 
11
 
12
  # from langchain.embeddings import HuggingFaceEmbeddings
13
 
 
5
  import qdrant_client
6
 
7
 
8
+ from sentence_transformers import SentenceTransformer, util
9
  #model = SentenceTransformer('paraphrase-multilingual-MiniLM-L12-v2') # good so far
10
+ model = SentenceTransformer('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2') # good so far
11
+
12
+
13
+ # Set the cache folder to the one you created
14
+ cache_folder = '/app/sentence_transformers_cache'
15
+ util.cache_folder = cache_folder
16
+
17
 
18
  # from langchain.embeddings import HuggingFaceEmbeddings
19