Update README.md

README.md

@@ -1,25 +1,21 @@
 ---
-model_name: Wheat Anomaly Detection Model (ONNX)
+model_name: Wheat Anomaly Detection Model
 tags:
-  - onnx
+  - pytorch
   - resnet
   - agriculture
   - anomaly-detection
-  - image-classification
-  - wheat-disease-detection
-  - pest-detection
-  - agricultural-ai
 license: apache-2.0
-library_name: onnx
+library_name: transformers
 datasets:
-  - wheat-dataset   # Replace with the actual dataset name if available
-model_type: resnet50
+  - 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: onnx
+framework: pytorch
 task: image-classification
 pipeline_tag: image-classification