Update README.md

README.md

@@ -1,15 +1,23 @@
 ---
 model_name: Wheat Anomaly Detection Model
 tags:
-- pytorch
-- resnet
-- agriculture
-- anomaly-detection
+  - pytorch
+  - resnet
+  - agriculture
+  - anomaly-detection
 license: apache-2.0
-library_name: pytorch
+library_name: transformers
+datasets:
+  - wheat-disease-dataset
+model_type: resnet
+preprocessing:
+  - resize: 256
+  - center_crop: 224
+  - normalize: [0.485, 0.456, 0.406]
+  - normalize_std: [0.229, 0.224, 0.225]
+framework: pytorch
+task: image-classification
 pipeline_tag: image-classification
-language:
-- en
 ---
 
 # Wheat Anomaly Detection Model