ESG_Analysis / model /config.py
Mawube's picture
Update model/config.py
9ad0434
raw
history blame contribute delete
235 Bytes
from model.utils import AttributeDict
config = {
"model_name": "facebook/bart-large-mnli",
"pipeline_name":"zero-shot-classification",
"labels":["governance", "social", "environmental"],
}
config = AttributeDict(config)