Update app.py

app.py

@@ -3,6 +3,7 @@ import streamlit as st
 import pandas as pd
 import numpy as np
 import plotly.express as px
+import plotly.graph_objects as go
 from sklearn.metrics.pairwise import paired_cosine_distances
 from sklearn.preprocessing import normalize
 from rolaser import RoLaserEncoder
@@ -109,6 +110,22 @@ def main():
     st.header('Cosine Distance Statistics')
     st.caption('*This box plot is interactive: Hover on the boxes to display values. Click on the legend items to filter models.*')
     fig = px.box(outputs, x='model', y='cos', color='model', boxmean='sd')
+    fig = go.Figure()
+    fig.add_trace(go.Box(
+        y=outputs[outputs['model']=='LASER']['cos'],
+        name='LASER',
+        boxmean='sd'
+    ))
+    fig.add_trace(go.Box(
+        y=outputs[outputs['model']=='RoLASER']['cos'],
+        name='RoLASER',
+        boxmean='sd'
+    ))
+    fig.add_trace(go.Box(
+        y=outputs[outputs['model']=='c-RoLASER']['cos'],
+        name='c-RoLASER',
+        boxmean='sd'
+    ))
     fig.update_xaxes(title_text='Model')
     fig.update_yaxes(title_text='Cosine Distance')
     st.plotly_chart(fig, use_container_width=True)