Rajor78 commited on
Commit
7bbf29e
verified
1 Parent(s): 9981e20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
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
- return resultados
 
 
 
 
 
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="dataframe"
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()