Update config.json

config.json

@@ -1,8 +1,15 @@
 {
-  "task": "image-classification",
-  "framework": "pytorch",
-  "model_id": "ai-image-detector",
-  "handler_type": "custom",
-  "handler_file": "handler.py",
-  "handler_class": "Handler"
+  "architectures": ["AIDetectorForImageClassification"],
+  "model_type": "vision-encoder-decoder",
+  "num_labels": 2,
+  "id2label": {
+    "0": "real",
+    "1": "ai-generated"
+  },
+  "label2id": {
+    "real": 0,
+    "ai-generated": 1
+  },
+  "torch_dtype": "float32",
+  "transformers_version": "4.30.0"
 }