leaf-large / README.md
baskra's picture
Update README.md
941c13f verified
metadata
library_name: transformers
tags:
  - food
  - environment
  - NLP
  - Eco-Score
  - products
  - multilingual
  - BERT
  - classification
  - Open Food Facts
  - climate
license: mit
datasets:
  - baskra/LEAF

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]