Update app.py

app.py

@@ -106,7 +106,7 @@ def main(myargs):
 		if st.session_state['input_type'] == 'File' and "embeddings_all" in st.session_state and st.session_state.embeddings_plot in ["2D", "3D"]:
 			indices = [x for x in range(st.session_state.data_df[st.session_state.input_column].values.shape[0])]
 			if "selected_indices" in st.session_state:
-				if len(st.session_state.selected_indices) >=4:
+				if len(st.session_state.selected_indices) >4:
 					l = st.session_state.selected_indices
 					l.sort()
 					indices = l
@@ -164,7 +164,7 @@ def main(myargs):
 					plot = plotpx.scatter(
 								df_embeddings, x='x', y='y',
 								color='severity', labels={'color': 'severity'},
-								hover_data=['text','all_predictions','data_index'],title = 'BERT Embeddings Visualization - Please select rows (at least 4) to display specific examples'
+								hover_data=['text','all_predictions','data_index'],title = 'BERT Embeddings Visualization - Please select rows (at least 5) to display specific examples'
 					)
 
 				else:
@@ -172,7 +172,7 @@ def main(myargs):
 					plot = plotpx.scatter_3d(
 								df_embeddings, x='x', y='y', z='z',
 								color='severity', labels={'color': 'severity'},
-								hover_data=['text','all_predictions','data_index'],title = 'BERT Embeddings Visualization - Please select rows (at least 4) to display specific examples'
+								hover_data=['text','all_predictions','data_index'],title = 'BERT Embeddings Visualization - Please select rows (at least 5) to display specific examples'
 					)
 
 				st.plotly_chart(plot,use_container_width=True,)