newsmediabias/FakeWatch

from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline

# Load model and tokenizer
tokenizer = AutoTokenizer.from_pretrained("newsmediabias/fake-news-classifier-elections")
model = AutoModelForSequenceClassification.from_pretrained("newsmediabias/fake-news-classifier-elections")

# Initialize the pipeline for sequence classification with the model and tokenizer
fake_news_classifier = pipeline(
    "text-classification",
    model=model,
    tokenizer=tokenizer
)

# Define label mapping based on the model's training
label_mapping = {
    "LABEL_0": "REAL",
    "LABEL_1": "FAKE"
}

# Example text to classify
example_text = "The election was rigged and full of fraud."

# Perform inference
predictions = fake_news_classifier(example_text)
predicted_class = label_mapping[predictions[0]['label']]
confidence_score = predictions[0]['score'] * 100

# Output the results
print(f"Predicted Class: {predicted_class}")
print(f"Confidence Score: {confidence_score:.2f}%")