Views
No views yet
TRUE vs FAKE).distilbert-base-uncasedsafetensors (268MB)0: TRUE (real news)1: FAKE (misleading content)1from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
2from huggingface_hub import snapshot_download
3import torch
4
5model_path = snapshot_download("Barhomy/distilbert-fake-news-model")
6tokenizer = DistilBertTokenizer.from_pretrained(model_path)
7model = DistilBertForSequenceClassification.from_pretrained(model_path)
8
9inputs = tokenizer("Aliens have landed in Paris!", return_tensors="pt")
10outputs = model(**inputs)
11prediction = torch.argmax(outputs.logits).item()
12print("FAKE" if prediction == 1 else "TRUE")