CristianR8 commited on
Commit
2ab6dd7
·
verified ·
1 Parent(s): c7427cf

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -3
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "num_labels": 6,
3
  "id2label": {
4
  "0": "fermentado",
5
  "1": "hongo",
@@ -18,5 +18,6 @@
18
  },
19
  "model_type": "resnet",
20
  "num_channels": 3,
21
- "transformers_version": "4.42.4"
22
- }
 
 
1
  {
2
+ "num_labels": 6,
3
  "id2label": {
4
  "0": "fermentado",
5
  "1": "hongo",
 
18
  },
19
  "model_type": "resnet",
20
  "num_channels": 3,
21
+ "transformers_version": "4.42.4",
22
+ "feature_extractor_type": "ImageFeatureExtractor"
23
+ }