Downloads · 30 days
8
1% of all-time downloads
nicotaroni/distilbert-multilingual-cased_fine_tuned_
distilbert-multilingual-cased_fine_tuned_ is a text classification model from nicotaroni. Use it when you need a label for a piece of text. It is set up for transformers.
The model has been fine-tuned for a multilingual classification text task: it recognizes whether a real-estate advertisement is an agency advertisement (label = 1) or a private advertisement (label = 0).
Downloads · 30 days
8
1% of all-time downloads
All-time downloads
830
Public
Repo size
5.4 GB
Likes
0
Public
Click a slice to open those files.
.bin541 MB · 99%
From the Hugging Face model README
The model has been fine-tuned for a multilingual classification text task: it recognizes whether a real-estate advertisement is an agency advertisement (label = 1) or a private advertisement (label = 0).
Use the code below to get started with the model:
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
tokenizer = AutoTokenizer.from_pretrained("nicotaroni/distilbert-multilingual-cased_fine_tuned_", max_length=512,truncation=True)
model = AutoModelForSequenceClassification.from_pretrained("nicotaroni/distilbert-multilingual-cased_fine_tuned_")
classifier = pipeline("text-classification", model=model, tokenizer=tokenizer, truncation = True)
text = " real estate advertisement "
outputs = classifier(text)
print(outputs)