Update model with proper task configuration

README.md

@@ -0,0 +1,24 @@
+---
+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 image, passed the safety check
+- Class 1: Fraudulent image, failed the safety check
+
+## Usage
+
+from transformers import pipeline
+
+classifier = pipeline("image-classification", model="IrishMehta/fraud-detection-idnet-two-class")
+result = classifier("path/to/image.jpg")
+print(result)
+

config.json

@@ -0,0 +1 @@
+{"architectures": ["ResNetForImageClassification"], "model_type": "resnet", "num_labels": 2, "id2label": {"0": "0", "1": "1"}, "label2id": {"0": 0, "1": 1}, "hidden_size": 2048, "problem_type": "single_label_classification", "pipeline_tag": "image-classification", "AutoModelForImageClassification": "transformers.models.resnet.modeling_resnet.ResNetForImageClassification"}

preprocessor_config.json

@@ -0,0 +1,22 @@
+{
+  "crop_pct": 0.875,
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.485,
+    0.456,
+    0.406
+  ],
+  "image_processor_type": "ConvNextImageProcessor",
+  "image_std": [
+    0.229,
+    0.224,
+    0.225
+  ],
+  "resample": 3,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "shortest_edge": 224
+  }
+}

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0dc9790c52014b9f2f91adc7b24b2afcadae02f0ecc6b86d8985b85cb6e04776
+size 94363594