nicotaroni/distilbert-multilingual-cased_fine_tuned_

Model Card for Model ID

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).

• Developed by: [nicotaroni]
• Model type: [text-classification]
• Language(s) (NLP): [Multilingual]
• Finetuned from model [optional]: [distilbert-multilingual-cased]

How to Get Started with the Model

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)