Update app.py
Browse files
app.py
CHANGED
@@ -1,36 +1,31 @@
|
|
1 |
-
|
2 |
-
|
|
|
|
|
3 |
|
4 |
-
# Importar
|
5 |
from sentence_transformers import SentenceTransformer
|
6 |
import pandas as pd
|
7 |
import chromadb
|
8 |
from chromadb.utils import embedding_functions
|
9 |
-
from scipy.spatial.distance import cosine
|
10 |
-
from tqdm import tqdm
|
11 |
from datetime import datetime
|
12 |
import gradio as gr
|
13 |
-
import os
|
14 |
|
15 |
# Configuraci贸n de variables
|
16 |
EMBEDDING_LLM = "jinaai/jina-embeddings-v2-base-es" # Modelo de embedding utilizado
|
17 |
|
18 |
# Cargar el modelo de embeddings
|
19 |
-
model = SentenceTransformer(
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
|
24 |
-
#
|
25 |
-
|
|
|
26 |
|
27 |
# Cargar la base de datos, especificando la codificaci贸n
|
28 |
-
df = pd.read_csv(
|
29 |
-
file_path,
|
30 |
-
index_col=0,
|
31 |
-
encoding='utf-8',
|
32 |
-
sep='|'
|
33 |
-
)
|
34 |
|
35 |
# Crear una columna combinada para embeddings
|
36 |
df['text'] = df.apply(
|
@@ -127,4 +122,4 @@ ui = gr.Interface(
|
|
127 |
)
|
128 |
|
129 |
# Ejecutar la aplicaci贸n
|
130 |
-
ui.launch(server_name="0.0.0.0", server_port=7860)
|
|
|
1 |
+
import os
|
2 |
+
|
3 |
+
# Instalar las dependencias necesarias
|
4 |
+
os.system("pip install -U sentence-transformers chromadb pandas gradio numpy scipy")
|
5 |
|
6 |
+
# Importar las bibliotecas necesarias
|
7 |
from sentence_transformers import SentenceTransformer
|
8 |
import pandas as pd
|
9 |
import chromadb
|
10 |
from chromadb.utils import embedding_functions
|
|
|
|
|
11 |
from datetime import datetime
|
12 |
import gradio as gr
|
|
|
13 |
|
14 |
# Configuraci贸n de variables
|
15 |
EMBEDDING_LLM = "jinaai/jina-embeddings-v2-base-es" # Modelo de embedding utilizado
|
16 |
|
17 |
# Cargar el modelo de embeddings
|
18 |
+
model = SentenceTransformer(EMBEDDING_LLM, trust_remote_code=True)
|
19 |
+
|
20 |
+
# Ruta del archivo CSV
|
21 |
+
file_path = "Diagrama_de_decision.csv" # Aseg煤rate de subir este archivo al espacio
|
22 |
|
23 |
+
# Verificar que el archivo existe
|
24 |
+
if not os.path.exists(file_path):
|
25 |
+
raise FileNotFoundError(f"El archivo {file_path} no se encontr贸. Por favor, s煤belo al directorio del Space.")
|
26 |
|
27 |
# Cargar la base de datos, especificando la codificaci贸n
|
28 |
+
df = pd.read_csv(file_path, index_col=0, encoding='utf-8', sep='|')
|
|
|
|
|
|
|
|
|
|
|
29 |
|
30 |
# Crear una columna combinada para embeddings
|
31 |
df['text'] = df.apply(
|
|
|
122 |
)
|
123 |
|
124 |
# Ejecutar la aplicaci贸n
|
125 |
+
ui.launch(server_name="0.0.0.0", server_port=7860)
|