MoritzMMuller commited on
Commit
731269d
·
verified ·
1 Parent(s): 1352bba

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +0 -7
config.json CHANGED
@@ -1,14 +1,7 @@
1
  {
2
- // 1) Tell HF which class to use (must match your modeling code)
3
  "architectures": ["ConvNextForImageClassification"],
4
-
5
- // 2) The key Transformers looks for:
6
  "model_type": "convnext",
7
-
8
- // 3) Your number of labels
9
  "num_labels": 2,
10
-
11
- // 4) (Optional) Helpful for pipelines
12
  "id2label": { "0": "benign", "1": "malignant" },
13
  "label2id": { "benign": 0, "malignant": 1 }
14
  }
 
1
  {
 
2
  "architectures": ["ConvNextForImageClassification"],
 
 
3
  "model_type": "convnext",
 
 
4
  "num_labels": 2,
 
 
5
  "id2label": { "0": "benign", "1": "malignant" },
6
  "label2id": { "benign": 0, "malignant": 1 }
7
  }