jacoballessio
/

ai-image-detect-distilled

Image Classification

real-fake-detection

ai-image-detection

Model card Files Files and versions Community

jacoballessio commited on Jul 1, 2024

Commit

028a63e

·

verified ·

1 Parent(s): b91e8fd

Update config.json

Files changed (1) hide show

config.json +5 -3

config.json CHANGED Viewed

@@ -4,7 +4,7 @@
     "EfficientNetForImageClassification"
   ],
   "model_type": "efficientnet",
-  "num_classes": 2,
   "id2label": {
     "0": "fake",
     "1": "real"
@@ -22,7 +22,9 @@
     "threshold": 0.58
   },
   "classifier_dropout_prob": 0.2,
-  "num_features": 1792,
   "hidden_act": "swish",
-  "transformers_version": "4.41.2"
 }

     "EfficientNetForImageClassification"
   ],
   "model_type": "efficientnet",
+  "num_labels": 2,
   "id2label": {
     "0": "fake",
     "1": "real"
     "threshold": 0.58
   },
   "classifier_dropout_prob": 0.2,
   "hidden_act": "swish",
+  "transformers_version": "4.41.2",
+  "torch_dtype": "float32",
+  "use_cache": true,
+  "output_hidden_states": false
 }