Update config.json

config.json

@@ -1 +1,22 @@
-{"architecture": "densenet121", "num_labels": 3}
+{
+    "_name_or_path": "densenet-finetuned",
+    "architectures": [
+        "DenseNetForImageClassification"
+    ],
+    "model_type": "image-classification",
+    "num_labels": 3,
+    "hidden_size": 1024,
+    "label2id": {
+        "Real": 0,
+        "AI": 1,
+        "Other": 2
+    },
+    "id2label": {
+        "0": "Real",
+        "1": "AI",
+        "2": "Other"
+    },
+    "torchvision_backbone": "densenet121",
+    "fine_tuned": true,
+    "framework": "pytorch"
+}