commit files to HF hub

config.json

@@ -0,0 +1,13 @@
+{
+  "DEVICE": "cpu",
+  "architectures": [
+    "DeepFakeModel"
+  ],
+  "auto_map": {
+    "AutoConfig": "deepfakeconfig.DeepFakeConfig",
+    "AutoModelForImageClassification": "deepfakemodel.DeepFakeModel"
+  },
+  "model_type": "ResNet",
+  "torch_dtype": "float32",
+  "transformers_version": "4.37.0"
+}

deepfakeconfig.py

@@ -0,0 +1,10 @@
+from transformers import PretrainedConfig
+import torch 
+class DeepFakeConfig(PretrainedConfig):
+    model_type = "ResNet"
+    def __init__(self,**kwargs):
+        super().__init__(**kwargs)
+        self.DEVICE = 'cuda:0' if torch.cuda.is_available() else 'cpu'
+
+config = DeepFakeConfig()
+config.register_for_auto_class()

deepfakemodel.py

@@ -0,0 +1,15 @@
+from transformers import  PreTrainedModel
+from facenet_pytorch import MTCNN, InceptionResnetV1
+from deepfakeconfig import DeepFakeConfig
+config = DeepFakeConfig()
+
+class DeepFakeModel(PreTrainedModel):
+    config_class = DeepFakeConfig
+    def __init__(self, config):
+        super().__init__(config)
+        self.model = InceptionResnetV1(
+            pretrained="vggface2",
+            classify=True,
+            num_classes=1,
+            device=config.DEVICE
+        )

model.safetensors

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

requirements.txt

@@ -0,0 +1,6 @@
+Pillow
+facenet-pytorch==2.5.2
+torch==1.11.0
+opencv-python
+grad-cam
+transformers