Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,6 +4,7 @@ from sklearn.cluster import DBSCAN
|
|
| 4 |
from sklearn.metrics.pairwise import cosine_similarity
|
| 5 |
import numpy as np
|
| 6 |
import gradio as gr
|
|
|
|
| 7 |
|
| 8 |
# Cargar el modelo de embeddings
|
| 9 |
modelo = SentenceTransformer('all-MiniLM-L6-v2')
|
|
@@ -63,7 +64,12 @@ def obtener_categorias_dinamicas(frases, umbral_similitud=0.5):
|
|
| 63 |
'Categor铆as': categorias_frases
|
| 64 |
})
|
| 65 |
|
| 66 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 67 |
|
| 68 |
# Funci贸n para cargar y procesar el archivo Excel
|
| 69 |
def procesar_excel(archivo):
|
|
@@ -76,7 +82,7 @@ def procesar_excel(archivo):
|
|
| 76 |
iface = gr.Interface(
|
| 77 |
fn=procesar_excel,
|
| 78 |
inputs=gr.File(label="Cargar archivo Excel con frases"),
|
| 79 |
-
outputs="
|
| 80 |
)
|
| 81 |
|
| 82 |
iface.launch()
|
|
|
|
| 4 |
from sklearn.metrics.pairwise import cosine_similarity
|
| 5 |
import numpy as np
|
| 6 |
import gradio as gr
|
| 7 |
+
import io
|
| 8 |
|
| 9 |
# Cargar el modelo de embeddings
|
| 10 |
modelo = SentenceTransformer('all-MiniLM-L6-v2')
|
|
|
|
| 64 |
'Categor铆as': categorias_frases
|
| 65 |
})
|
| 66 |
|
| 67 |
+
# Guardar los resultados en un archivo Excel en memoria (para devolverlo en Gradio)
|
| 68 |
+
output = io.BytesIO() # Buffer en memoria para escribir el archivo Excel
|
| 69 |
+
resultados.to_excel(output, index=False, encoding='utf-8')
|
| 70 |
+
output.seek(0) # Volver al principio del archivo para que Gradio pueda leerlo
|
| 71 |
+
|
| 72 |
+
return output
|
| 73 |
|
| 74 |
# Funci贸n para cargar y procesar el archivo Excel
|
| 75 |
def procesar_excel(archivo):
|
|
|
|
| 82 |
iface = gr.Interface(
|
| 83 |
fn=procesar_excel,
|
| 84 |
inputs=gr.File(label="Cargar archivo Excel con frases"),
|
| 85 |
+
outputs=gr.File(label="Descargar archivo Excel codificado") # Cambiar el output a archivo
|
| 86 |
)
|
| 87 |
|
| 88 |
iface.launch()
|