my1-deepfake-detector / models /text_model_1.py
ash12321's picture
Rename models/text_model.py to models/text_model_1.py
259d36d verified
raw
history blame contribute delete
286 Bytes
from transformers import pipeline
# Public fake news detection model
text_model = pipeline(
"text-classification",
model="therealcyberlord/fake-news-classification-distilbert"
)
# Label mapping for readability
text_label_map = {
"0": "Fake News",
"1": "Real News"
}