from transformers import AutoModelForSequenceClassification, AutoTokenizer
import torch

model_path = "your-huggingface-username/newsguard-ai-fake-news"
tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForSequenceClassification.from_pretrained(model_path)

text = "Some news article text here..."
inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True, max_length=512)

with torch.no_grad():
    outputs = model(**inputs)
    probs = torch.nn.functional.softmax(outputs.logits, dim=-1)

prediction = "Fake" if torch.argmax(probs) == 0 else "Real"
print(f"Prediction: {prediction}, Confidence: {probs.tolist()[0]}")

nallarahul/NewsGaurd

NewsGuard AI - Fake News Detection Model

Model Details

How to Use

nallarahul/
NewsGaurd