You can check the model's fine-tuning code on my GitHub.
FakeBerta is a fine-tuned version of DistilRoBERTa for detecting fake news. The model is trained to classify news articles as real (0) or fake (1) using natural language processing (NLP) techniques.
Base Model: DistilRoBERTa
Task: Fake news classification
1from transformers import AutoModelForSequenceClassification, AutoTokenizer
2import torch
3
4model_name = "YerayEsp/FakeBerta"
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForSequenceClassification.from_pretrained(model_name)
7
8inputs = tokenizer("Breaking: Scientists discover water on Mars!", return_tensors="pt")
9outputs = model(**inputs)
10
11logits = outputs.logits
12predicted_class = torch.argmax(logits).item()
13
14print(f"Predicted class: {predicted_class}") # 0 = Real, 1 = Fake
15