shulik7's picture
Upload config.json with huggingface_hub
805064d verified
{
"model_type": "bert",
"task_specific_params": {
"text-classification": {
"model": "Rostlab/prot_bert_bfd_localization"
}
},
"id2label": {
"0": "Cell.membrane",
"1": "Cytoplasm",
"2": "Endoplasmic.reticulum",
"3": "Extracellular",
"4": "Golgi.apparatus",
"5": "Lysosome/Vacuole",
"6": "Mitochondrion",
"7": "Nucleus",
"8": "Peroxisome",
"9": "Plastid"
},
"label2id": {
"Cell.membrane": 0,
"Cytoplasm": 1,
"Endoplasmic.reticulum": 2,
"Extracellular": 3,
"Golgi.apparatus": 4,
"Lysosome/Vacuole": 5,
"Mitochondrion": 6,
"Nucleus": 7,
"Peroxisome": 8,
"Plastid": 9
},
"num_labels": 10,
"transformers_version": "4.57.1"
}