aymanemalih commited on
Commit
e838163
·
1 Parent(s): 30f325c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -1
main.py CHANGED
@@ -3,12 +3,15 @@ from qdrant_client.http import models
3
  from flask_cors import CORS
4
  import openai
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'
 
3
  from flask_cors import CORS
4
  import openai
5
  import qdrant_client
6
+ import os
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
+ # Set the environment variable TRANSFORMERS_CACHE to the writable directory
13
+ os.environ['TRANSFORMERS_CACHE'] = '/app/sentence_transformers_cache'
14
+
15
 
16
  # Set the cache folder to the one you created
17
  cache_folder = '/app/sentence_transformers_cache'