# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("tomhaishiwo/conflicbert_binary_1")
model = AutoModelForSequenceClassification.from_pretrained("tomhaishiwo/conflicbert_binary_1", device_map="auto")Quick Links
This is a fine-tune model based on snowood1/ConfliBERT-scr-uncased, the dataset used is 20news with 8800 training binary labeled data. Please refer to author's original paper : https://github.com/eventdata/ConfliBERT
- Downloads last month
- 4
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="tomhaishiwo/conflicbert_binary_1")