Update src/streamlit_app.py

src/streamlit_app.py

@@ -643,6 +643,7 @@ if run_button:
                 df = pd.DataFrame(all_entities)
 
                 if not df.empty:
+                    df = df.reset_index(drop=True)
                     if st.session_state.is_custom_mode:
                         df['category'] = 'User Defined Entities'
                     else: