baskra/leaf-large

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

The leaf-large model is a BAAI/bge-m3 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([[-12.2842, ...]]), '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]