Update model with proper task configuration

README.md

@@ -0,0 +1,25 @@
+---
+language: en
+license: mit
+tags:
+- image-classification
+- computer-vision
+- fraud-detection
+pipeline_tag: image-classification
+---
+
+# Fraud Detection Image Classification Model
+
+This model classifies images into 3 classes:
+- Class 0: Non fraudulent images, passed the safety check
+- Class 1: Fraudulent images of type 5- Inpaint and Rewriting found in the ID
+- Class 2: Fraudulent images of type 6- Cropping and replacement found in the ID
+
+## Usage
+
+from transformers import pipeline
+
+classifier = pipeline("image-classification", model="IrishMehta/fraud-detection-idnet-three-class")
+result = classifier("path/to/image.jpg")
+print(result)
+

config.json

@@ -1,14 +1 @@
-{
-  "hidden_size": 2048,
-  "id2label": {
-    "0": "0",
-    "1": "1",
-    "2": "2"
-  },
-  "label2id": {
-    "0": 0,
-    "1": 1,
-    "2": 2
-  },
-  "transformers_version": "4.45.2"
-}
+{"architectures": ["ResNetForImageClassification"], "model_type": "resnet", "num_labels": 3, "id2label": {"0": "0", "1": "1", "2": "2"}, "label2id": {"0": 0, "1": 1, "2": 2}, "hidden_size": 2048, "problem_type": "single_label_classification", "pipeline_tag": "image-classification"}