baskra/leaf-base

LEAF: Predicting the Environmental Impact of Food Products based on their Name

The leaf-base model is a distiluse-base-multilingual-cased-v2 model fine-tuned on the LEAF dataset.

To load the model, use the following code:

from transformers import AutoTokenizer, AutoModel

tokenizer = AutoTokenizer.from_pretrained("baskra/leaf-base")
model = AutoModel.from_pretrained("baskra/leaf-base", trust_remote_code=True)

model(**tokenizer("Nutella", return_tensors="pt"))
# {'logits': tensor([[-11.9081, ...]]), 'class_idx': tensor([1553]), 'ef_score': tensor([0.0129]), 'class': ['Chocolate spread with hazelnuts']}

Citation

When using this model, please consider citing it as follows:

BibTeX:

[More Information Needed]

APA:

[More Information Needed]