Merge branch 'main' of https://huggingface.co/spaces/de-Rodrigo/Embeddings

app.py

@@ -1285,39 +1285,6 @@ def run_model(model_name):
                     
                     # Mostrar el heatmap en la app
                     st.bokeh_chart(heatmap_fig)
-                    
-                    # # Guardar la figura y el nombre correspondiente
-                    # heatmap_figures.append(heatmap_fig)
-                    # heatmap_names.append(f"heatmap_{selected_feature}_{x_comp}_vs_{y_comp}.png")
-
-
-                    # # Inicializar el driver de Chrome en modo headless
-                    # chrome_options = Options()
-                    # chrome_options.add_argument("--headless")
-                    # chrome_options.add_argument("--no-sandbox")
-                    # chrome_options.add_argument("--disable-dev-shm-usage")
-                    # driver = webdriver.Chrome(options=chrome_options, service=ChromeService(ChromeDriverManager().install()))
-
-                    # if st.button("Descargar todos los heatmaps"):
-                    #     zip_buffer = io.BytesIO()
-                    #     with tempfile.TemporaryDirectory() as tmpdirname:
-                    #         with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zip_file:
-                    #             # Exporta cada figura a PNG y añádela al ZIP
-                    #             for fig, filename in zip(heatmap_figures, heatmap_names):
-                    #                 tmp_path = f"{tmpdirname}/{filename}"
-                    #                 export_png(fig, filename=tmp_path, driver=driver)
-                    #                 with open(tmp_path, "rb") as f:
-                    #                     zip_file.writestr(filename, f.read())
-                    #     zip_buffer.seek(0)
-                    #     st.download_button(
-                    #         label="Descargar ZIP de heatmaps",
-                    #         data=zip_buffer,
-                    #         file_name="heatmaps.zip",
-                    #         mime="application/zip",
-                    #         key=f"download_heatmaps_{model_name}"
-                    #     )
-
-                    # driver.quit()
 
 
 def main():