Update app.py

app.py

@@ -92,8 +92,8 @@ words_df = pd.DataFrame(
     st.session_state['words'],
     columns=["word", "similarity"]
 ).sort_values(by=["similarity"], ascending=False)
-st.dataframe(words_df)
+st.dataframe(words_df, use_container_width=True)
 
 
-fig_3d = px.scatter_3d(word_points, x="x", y="y", z="z", color="similarity", hover_name="l", hover_data={"x": False, "y": False, "z": False, "s": False}, size="s", size_max=10, range_color=(0,1))
+fig_3d = px.scatter_3d(words_umap_df, x="x", y="y", z="z", color="similarity", hover_name="l", hover_data={"x": False, "y": False, "z": False, "s": False}, size="s", size_max=10, range_color=(0,1))
 st.plotly_chart(fig_3d, theme="streamlit", use_container_width=True)