Update app.py

app.py

@@ -130,8 +130,8 @@ if st.button("Sentiment Analysis", type="secondary"):
                         results = []
                         for i, label in enumerate(predicted_labels):
                             results.append({'Review Number': i + 1, 'Sentiment': model.config.id2label[label.item()]})
-                        sentiment_df1 = pd.DataFrame(results)
-                        sentiment_df = sentiment_df1.drop(sentiment_df.columns[['Review Number']], axis=1)
+                        sentiment_df = pd.DataFrame(results)
+                    
 
 
                         
@@ -155,7 +155,9 @@ if st.button("Sentiment Analysis", type="secondary"):
                         plt.imshow(wc)
                         plt.axis('off')
                         st.pyplot(fig)
-                    csv = result.to_csv(index=False)
+                    
+                    result1 = result.drop(result.columns[['Review Number']], axis=1)
+                    csv = result1.to_csv(index=False)
                     st.download_button(
                         label="Download data as CSV",
                         data=csv,