Create config.json

config.json

@@ -0,0 +1,21 @@
+{
+  "_name_or_path": "Skarn55/ai_detection",
+  "architectures": ["ImageClassification"],
+  "model_type": "vision",
+  "task_specific_params": {
+    "image-classification": {
+      "id2label": {
+        "0": "real",
+        "1": "ai_generated",
+        "2": "deepfake"
+      },
+      "label2id": {
+        "real": 0,
+        "ai_generated": 1,
+        "deepfake": 2
+      }
+    }
+  },
+  "num_labels": 3,
+  "image_size": 224
+}