Update app.py

app.py

@@ -100,7 +100,7 @@ def main():
     outputs['cos'] = np.concatenate([X_cos_laser, X_cos_rolaser, X_cos_c_rolaser])
 
     st.header('Cosine Distance Scores')
-    st.caption('*This bar plot is interactive: Hover on the bars to display values. Click on the legend items to filter models.*')
+    st.caption('*This box plot is interactive: Hover on the boxes to display values. Click on the legend items to filter models.*')
     fig = px.bar(outputs, x='pair', y='cos', color='model', barmode='group', hover_data=['ugc', 'std'])
     fig.update_xaxes(title_text='Text Input Pair')
     fig.update_yaxes(title_text='Cosine Distance')
@@ -108,7 +108,10 @@ def main():
 
     st.header('Average Cosine Distance Scores')
     st.caption('*This data table is interactive: Click on a column header to sort values.*')
-    st.write(outputs.groupby('model')['cos'].describe().sort_values(by='mean', ascending=True))
-
+    fig = px.box(outputs, x='model', y='cos', color='model')
+    fig.update_xaxes(title_text='Model')
+    fig.update_yaxes(title_text='Cosine Distance')
+    st.plotly_chart(fig, use_container_width=True)
+    
 if __name__ == "__main__":
     main()