Mauro24 commited on
Commit
d1ca29f
·
verified ·
1 Parent(s): 4901dcb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -7,12 +7,12 @@ import os
7
 
8
  # Percorsi per il primo file ZIP
9
  zip_path_m = "faiss_manual_index.zip" # File ZIP per l'indice manuale
10
- extraction_dir_m = "./extracted_models/manual_index" # Sottocartella per estrazione manuale
11
  testm_dir = "./extracted_models/faiss_manual_index" # Cartella finale
12
 
13
  # Percorsi per il secondo file ZIP
14
  zip_path_p = "faiss_problems_index.zip" # File ZIP per l'indice problemi
15
- extraction_dir_p = "./extracted_models/problems_index" # Sottocartella per estrazione problemi
16
  testp_dir = "./extracted_models/faiss_problems_index" # Cartella finale
17
 
18
  # Estrai il primo file ZIP se non esiste già
@@ -38,8 +38,8 @@ embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/LaBSE"
38
 
39
  # Carica i vectorstore FAISS salvati
40
 
41
- manual_vectorstore = FAISS.load_local("faiss_manual_index", embedding_model, allow_dangerous_deserialization=True)
42
- problems_vectorstore = FAISS.load_local("faiss_problems_index", embedding_model, allow_dangerous_deserialization=True)
43
 
44
  def search_query(query):
45
  # Cerca nei manuali
 
7
 
8
  # Percorsi per il primo file ZIP
9
  zip_path_m = "faiss_manual_index.zip" # File ZIP per l'indice manuale
10
+ extraction_dir_m = "./extracted_models/faiss_manual_index" # Sottocartella per estrazione manuale
11
  testm_dir = "./extracted_models/faiss_manual_index" # Cartella finale
12
 
13
  # Percorsi per il secondo file ZIP
14
  zip_path_p = "faiss_problems_index.zip" # File ZIP per l'indice problemi
15
+ extraction_dir_p = "./extracted_models/faiss_problems_index" # Sottocartella per estrazione problemi
16
  testp_dir = "./extracted_models/faiss_problems_index" # Cartella finale
17
 
18
  # Estrai il primo file ZIP se non esiste già
 
38
 
39
  # Carica i vectorstore FAISS salvati
40
 
41
+ manual_vectorstore = FAISS.load_local(testm_dir, embedding_model, allow_dangerous_deserialization=True)
42
+ problems_vectorstore = FAISS.load_local(testp_dir, embedding_model, allow_dangerous_deserialization=True)
43
 
44
  def search_query(query):
45
  # Cerca nei manuali