File size: 235 Bytes
8d39dda
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
from model.utils import AttributeDict

config = {
    "model_name": "facebook/bart-large-mnli",
    "pipeline_name":"zero-shot-classification", 
    "labels":["governance", "social", "environmental"],
}

config = AttributeDict(config)