jays009 commited on
Commit
9faeffd
·
verified ·
1 Parent(s): dcaa06f

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +14 -3
config.json CHANGED
@@ -1,7 +1,18 @@
1
  {
2
- "architectures": ["ResNet50"],
3
  "model_type": "resnet",
4
- "input_ids": "image",
5
  "task": "image-classification",
6
- "framework": "pytorch"
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  }
 
1
  {
 
2
  "model_type": "resnet",
 
3
  "task": "image-classification",
4
+ "framework": "pytorch",
5
+ "pipeline_tag": "image-classification",
6
+ "preprocessing": {
7
+ "resize": 256,
8
+ "center_crop": 224,
9
+ "normalize": [0.485, 0.456, 0.406],
10
+ "normalize_std": [0.229, 0.224, 0.225]
11
+ },
12
+ "license": "apache-2.0",
13
+ "tags": [
14
+ "image-classification",
15
+ "wheat-disease-detection",
16
+ "agriculture"
17
+ ]
18
  }