Mauro24 commited on
Commit
0deea66
·
verified ·
1 Parent(s): 797ac0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +30 -0
app.py CHANGED
@@ -2,7 +2,37 @@
2
  import gradio as gr
3
  from langchain.vectorstores import FAISS
4
  from langchain.embeddings import HuggingFaceEmbeddings
 
5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  # Carica il modello di embedding
7
  embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/LaBSE")
8
 
 
2
  import gradio as gr
3
  from langchain.vectorstores import FAISS
4
  from langchain.embeddings import HuggingFaceEmbeddings
5
+ import zipfile
6
 
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/manual_index/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/problems_index/faiss_problems_index" # Cartella finale
17
+
18
+ # Estrai il primo file ZIP se non esiste già
19
+ if not os.path.exists(testm_dir):
20
+ with zipfile.ZipFile(zip_path_m, 'r') as zip_ref:
21
+ zip_ref.extractall(extraction_dir_m)
22
+ print(f"Indice Manuale estratto correttamente nella cartella {extraction_dir_m}")
23
+ else:
24
+ print(f"Indice Manuale già presente in {testm_dir}")
25
+
26
+ # Estrai il secondo file ZIP se non esiste già
27
+ if not os.path.exists(testp_dir):
28
+ with zipfile.ZipFile(zip_path_p, 'r') as zip_ref:
29
+ zip_ref.extractall(extraction_dir_p)
30
+ print(f"Indice Problemi estratto correttamente nella cartella {extraction_dir_p}")
31
+ else:
32
+ print(f"Indice Problemi già presente in {testp_dir}")
33
+
34
+
35
+
36
  # Carica il modello di embedding
37
  embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/LaBSE")
38