Spaces:
Running
Running
| from pathlib import Path | |
| import streamlit as st | |
| import streamlit.components.v1 as components | |
| import plotly.figure_factory as ff | |
| import numpy as np | |
| import pandas as pd | |
| #from streamlit_plotly_events import plotly_events | |
| def app(): | |
| st.markdown('#### Visualization') | |
| img_2d_embed = pd.read_csv('data/img_2d_embedding.csv', index_col=0) | |
| img_2d_embed = img_2d_embed.sample(frac=0.1, random_state=0) | |
| txt_2d_embed = pd.read_csv('data/txt_2d_embedding.csv', index_col=0) | |
| txt_2d_embed = txt_2d_embed.sample(frac=0.1, random_state=0) | |
| col1, col2 = st.columns(2) | |
| with col1: | |
| fig1 = ff.create_2d_density( | |
| x=img_2d_embed['UMAP_1'], | |
| y=img_2d_embed['UMAP_2'], | |
| #colors=img_2d_embed['tag'], | |
| colorscale='Blues', # set the color map | |
| height=500, # set height of the figure | |
| width=500, # set width of the figure | |
| title='Image embedding visualized in 2D UMAP' | |
| ) | |
| #selected_points = plotly_events(fig1, click_event=True, hover_event=True) | |
| st.plotly_chart(fig1, use_container_width=True) | |
| with col2: | |
| fig2 = ff.create_2d_density( | |
| x=txt_2d_embed['UMAP_1'], | |
| y=txt_2d_embed['UMAP_2'], | |
| #colors=img_2d_embed['tag'], | |
| colorscale='Blues', # set the color map | |
| height=500, # set height of the figure | |
| width=500, # set width of the figure | |
| title='Text embedding visualized in 2D UMAP' | |
| ) | |
| st.plotly_chart(fig2, use_container_width=True) |