---
pipeline_tag: zero-shot-classification
---
# Install transformers library
!pip install transformers

# Import necessary libraries
import pandas as pd
from transformers import pipeline
from google.colab import files

# Upload the file
uploaded = files.upload()

# Load the dataset
file_name = 'publications.csv'  # Use the file name as uploaded
df = pd.read_csv(file_name)

# Display the first few rows of the dataset
df.head()

# Load the zero-shot classification pipeline
classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")

# Define the categories
candidate_labels = ["World", "Sports", "Business", "Science and Technology", "Entertainment", "Lifestyle"]

# Classify each news article
results = []

for article in df['headline']:
    result = classifier(article, candidate_labels)
    category = result['labels'][0]
    results.append(category)

# Add the results to the DataFrame
df['category'] = results

# Display the categorized DataFrame
df.head(10)