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]
- Downloads last month
- 3
Inference API (serverless) does not yet support model repos that contain custom code.