Update app.py

app.py

@@ -112,7 +112,7 @@ with tab1:
         df = pd.read_csv(uploaded_file)
     
         # Load the fine-tuned news classifier
-        classifier = pipeline("text-classification", model="Prageeth-1/News_classification.2")
+        classifier = pipeline("text-classification", model="Imasha17/News_classification.3")
 
         # Preprocess
         # Lowercase
@@ -206,7 +206,7 @@ with tab1:
         df["Class"] = df["preprocessed_content"].apply(lambda text: classifier(text)[0]["label"])
 
         #Delete Unnecessary columns
-        df = df[['content','preprocessed_content','Class']]
+        df = df[['content','Class']]
 
                                
         # Show results
@@ -217,10 +217,6 @@ with tab1:
         st.subheader("Class Distribution")
         class_dist = df['Class'].value_counts()
         st.bar_chart(class_dist)
-                    
-
-        #Delete Unnecessary columns
-        df = df[['content','Class']]
         
                     
        # Download button