Update config.json

config.json

@@ -1,8 +1,52 @@
 {
-  "model_type": "custom",
-  "model_name": "dinov2-svm-embeddings-currency_indian",
-  "model_version": "1.0",
-  "description": "A model for classifying Indian currency notes using ViT and SVM.",
-  "input_size": 224,
-  "output_labels": ["1", "2", "5", "10", "20", "50", "100", "200", "500", "2000"]
+  "_name_or_path": "pjdevelop/dinov2-svm-embeddings-currency_indian",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "1",
+    "1": "10",
+    "2": "100",
+    "3": "2",
+    "4": "20",
+    "5": "200",
+    "6": "2000",
+    "7": "5",
+    "8": "50",
+    "9": "500",
+    "10": "Coin",
+    "11": "Rupee"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "1": 0,
+    "10": 1,
+    "100": 2,
+    "2": 3,
+    "20": 4,
+    "200": 5,
+    "2000": 6,
+    "5": 7,
+    "50": 8,
+    "500": 9,
+    "Coin": 10,
+    "Rupee": 11
+  },
+  "layer_norm_eps": 1e-12,
+  "model_type": "vit",
+  "num_attention_heads": 12,
+  "num_channels": 3,
+  "num_hidden_layers": 12,
+  "patch_size": 16,
+  "problem_type": "single_label_classification",
+  "qkv_bias": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.28.1"
 }