Spaces:
Running
Running
File size: 1,558 Bytes
7ee7b26 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
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) |