commit files to HF hub

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+__pycache__/deepfakemodel.cpython-39.pyc filter=lfs diff=lfs merge=lfs -text

__pycache__/deepfakemodel.cpython-39.pyc

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

deepfakeconfig.py

@@ -4,7 +4,4 @@ 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()
+        self.DEVICE = 'cuda:0' if torch.cuda.is_available() else 'cpu'

deepfakemodel.py

@@ -1,7 +1,6 @@
 from transformers import  PreTrainedModel
 from facenet_pytorch import MTCNN, InceptionResnetV1
 from deepfakeconfig import DeepFakeConfig
-config = DeepFakeConfig()
 
 class DeepFakeModel(PreTrainedModel):
     config_class = DeepFakeConfig
@@ -12,4 +11,8 @@ class DeepFakeModel(PreTrainedModel):
             classify=True,
             num_classes=1,
             device=config.DEVICE
-        )
+        )
+
+
+DeepFakeConfig.register_for_auto_class()
+DeepFakeModel.register_for_auto_class("AutoModelForImageClassification")