aymanemalih commited on
Commit
64f6d2d
·
1 Parent(s): 2ec25ce

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +10 -10
main.py CHANGED
@@ -5,28 +5,28 @@ 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
- model = SentenceTransformer('vectorizing_model')
13
 
14
- # Set the environment variable TRANSFORMERS_CACHE to the writable directory
15
- os.environ['TRANSFORMERS_CACHE'] = '/app/sentence_transformers_cache'
16
 
17
 
18
- # Set the cache folder to the one you created
19
- cache_folder = '/app/sentence_transformers_cache'
20
- util.cache_folder = cache_folder
21
 
22
 
23
  # from langchain.embeddings import HuggingFaceEmbeddings
24
 
25
  # model = HuggingFaceEmbeddings(model_name='paraphrase-multilingual-MiniLM-L12-v2')
26
 
27
- import os
28
- os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'
29
- os.environ['HF_HOME'] = '/blabla/cache/'
30
 
31
 
32
  start_message = 'Joue le Rôle d’un expert fiscale au Canada. Les réponses que tu va me fournir seront exploité par une API. Ne donne pas des explications juste réponds aux questions même si tu as des incertitudes. Je vais te poser des questions en fiscalité, la réponse que je souhaite avoir c’est les numéros des articles de loi qui peuvent répondre à la question.Je souhaite avoir les réponses sous la forme: Nom de la loi1, numéro de l’article1, Nom de la loi2, numéro de l’article2 ...'
 
5
  import qdrant_client
6
  import os
7
 
8
+ from sentence_transformers import SentenceTransformer
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
+ model = SentenceTransformer('./vectorizing_model')
13
 
14
+ # # # Set the environment variable TRANSFORMERS_CACHE to the writable directory
15
+ # # os.environ['TRANSFORMERS_CACHE'] = '/app/sentence_transformers_cache'
16
 
17
 
18
+ # # Set the cache folder to the one you created
19
+ # cache_folder = '/app/sentence_transformers_cache'
20
+ # util.cache_folder = cache_folder
21
 
22
 
23
  # from langchain.embeddings import HuggingFaceEmbeddings
24
 
25
  # model = HuggingFaceEmbeddings(model_name='paraphrase-multilingual-MiniLM-L12-v2')
26
 
27
+ # import os
28
+ # os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'
29
+ # os.environ['HF_HOME'] = '/blabla/cache/'
30
 
31
 
32
  start_message = 'Joue le Rôle d’un expert fiscale au Canada. Les réponses que tu va me fournir seront exploité par une API. Ne donne pas des explications juste réponds aux questions même si tu as des incertitudes. Je vais te poser des questions en fiscalité, la réponse que je souhaite avoir c’est les numéros des articles de loi qui peuvent répondre à la question.Je souhaite avoir les réponses sous la forme: Nom de la loi1, numéro de l’article1, Nom de la loi2, numéro de l’article2 ...'