How to use WealthFromAI/empire-content-distilbert with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="WealthFromAI/empire-content-distilbert")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("WealthFromAI/empire-content-distilbert") model = AutoModelForSequenceClassification.from_pretrained("WealthFromAI/empire-content-distilbert")
c7f2711 f368a70
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
{ "id2label": { "0": "smart_home", "1": "lifestyle", "2": "spiritual", "3": "ai", "4": "reviews" }, "label2id": { "smart_home": 0, "lifestyle": 1, "spiritual": 2, "ai": 3, "reviews": 4 } }
