Update app.py

app.py

@@ -156,7 +156,8 @@ if st.button("Sentiment Analysis", type="secondary"):
                         plt.axis('off')
                         st.pyplot(fig)
                     
-                    result1 = result.drop(result.columns['Review Number'], axis=1)
+                    
+                    result1 = result.drop('Review Number', axis=1)
                     csv = result1.to_csv(index=False)
                     st.download_button(
                         label="Download data as CSV",
@@ -169,5 +170,7 @@ if st.button("Sentiment Analysis", type="secondary"):
     else:
         st.warning(f"You have reached the maximum URL attempts ({max_attempts}).")
 
+st.divider()
+
 if 'url_count' in st.session_state:
     st.write(f"URL pasted {st.session_state['url_count']} times.")