Update config.json

config.json

@@ -1,25 +1,24 @@
 {
-  "model_name": "Maize Anomaly Detection Model",
+  "model_name": "Wheat Anomaly Detection Model (TensorFlow)",
   "tags": [
-    "pytorch",
+    "tensorflow",
     "resnet",
     "agriculture",
     "anomaly-detection"
   ],
   "license": "apache-2.0",
-  "library_name": "pytorch",
+  "library_name": "tensorflow",
   "datasets": [
-    "your_huggingface_username/your_dataset_name"
+    "wheat-dataset"   // Replace with actual dataset name if available
   ],
-  "architecture": "ResNet50",
-  "input_shape": [3, 224, 224],
-  "num_classes": 2,
-  "optimizer": "Adam",
-  "learning_rate": 0.001,
-  "epochs": 20,
-  "batch_size": 32,
-  "pretrained": false,
-  "finetuned": true,
-  "training_framework": "PyTorch",
-  "description": "This model is trained to detect anomalies in maize crops, such as pest infections, diseases, or nutrient deficiencies, using ResNet50 architecture."
+  "model_type": "resnet50",
+  "preprocessing": {
+    "resize": 256,
+    "center_crop": 224,
+    "normalize": [0.485, 0.456, 0.406],
+    "normalize_std": [0.229, 0.224, 0.225]
+  },
+  "framework": "tensorflow",
+  "task": "image-classification",
+  "pipeline_tag": "image-classification"
 }