Spaces:
Sleeping
Sleeping
Update config.json
Browse files- 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 |
}
|