Update djezzy.py
Browse files
djezzy.py
CHANGED
@@ -20,6 +20,9 @@ import shutil
|
|
20 |
import tempfile
|
21 |
index_source='index.faiss'
|
22 |
hh_source='index.pkl'
|
|
|
|
|
|
|
23 |
with tempfile.TemporaryDirectory() as temp_dir:
|
24 |
# Chemins des fichiers cibles dans le répertoire temporaire
|
25 |
index_target = os.path.join(temp_dir, 'index.faiss')
|
@@ -32,9 +35,7 @@ with tempfile.TemporaryDirectory() as temp_dir:
|
|
32 |
# Charger la base de données FAISS à partir du répertoire temporaire
|
33 |
vector_db = FAISS.load_local(temp_dir, embedding_llm, allow_dangerous_deserialization=True)
|
34 |
|
35 |
-
model_name = "sentence-transformers/all-MiniLM-L6-v2"
|
36 |
|
37 |
-
embedding_llm = SentenceTransformerEmbeddings(model_name=model_name)
|
38 |
|
39 |
def load_data(text_filename='docs_text.json', embeddings_filename='docs_embeddings.json'):
|
40 |
|
|
|
20 |
import tempfile
|
21 |
index_source='index.faiss'
|
22 |
hh_source='index.pkl'
|
23 |
+
model_name = "sentence-transformers/all-MiniLM-L6-v2"
|
24 |
+
|
25 |
+
embedding_llm = SentenceTransformerEmbeddings(model_name=model_name)
|
26 |
with tempfile.TemporaryDirectory() as temp_dir:
|
27 |
# Chemins des fichiers cibles dans le répertoire temporaire
|
28 |
index_target = os.path.join(temp_dir, 'index.faiss')
|
|
|
35 |
# Charger la base de données FAISS à partir du répertoire temporaire
|
36 |
vector_db = FAISS.load_local(temp_dir, embedding_llm, allow_dangerous_deserialization=True)
|
37 |
|
|
|
38 |
|
|
|
39 |
|
40 |
def load_data(text_filename='docs_text.json', embeddings_filename='docs_embeddings.json'):
|
41 |
|