Update model with proper task configuration

.ipynb_checkpoints/README-checkpoint.md

@@ -0,0 +1,24 @@
+---
+license: mit
+library_name: transformers
+pipeline_tag: image-classification
+tags:
+  - image-classification
+  - computer-vision
+---
+
+# 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 +1 @@
-{"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"}
+{"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", "AutoModelForImageClassification": "transformers.models.resnet.modeling_resnet.ResNetForImageClassification"}