Downloads · 30 days
7
7% of all-time downloads
Beelow/XLM_RoBERTa-Multilingual-Clickbait-Detection-RU-UA-Finetune
XLM_RoBERTa-Multilingual-Clickbait-Detection-RU-UA-Finetune is a text classification model from Beelow. Use it when you need a label for a piece of text. The card lists the license as mit.
Модель для детекції клікбейту та реклами в українських новинних заголовках. Базована на XLM-RoBERTa Large, дообучена на українському корпусі.
Downloads · 30 days
7
7% of all-time downloads
All-time downloads
96
Public
Parameters
560M
13.5 GB on disk
Likes
0
Public
Click a slice to open those files.
.pt9 GB · 57%
From the Hugging Face model README
Модель для детекції клікбейту та реклами в українських новинних заголовках. Базована на XLM-RoBERTa Large, дообучена на українському корпусі.
Ця модель класифікує новинні заголовки на два класи:
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
model_name = "YOUR_USERNAME/clickbait-detection-ukrainian"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)
def detect_clickbait(text: str, threshold: float = 0.7) -> dict:
inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512)
with torch.no_grad():
outputs = model(**inputs)
probs = torch.softmax(outputs.logits, dim=-1)
score = probs.item()[^13]
return {
"text": text,
"clickbait_score": score,
"is_clickbait": score >= threshold
}
# Приклади
headlines = [
"Шокуючі подробиці ДТП: ти не повіриш!",
"ЗСУ просунулися на східному фронті - Генштаб",
"Знижки -70%! Поспішай!"
]
for headline in headlines:
result = detect_clickbait(headline)
print(f"{result['clickbait_score']:.2f} - {headline}")
MIT License