jacoballessio commited on
Commit
4b7f212
·
verified ·
1 Parent(s): f491a47

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_name_or_path": "jacoballessio/ai-image-detect-distilled",
3
  "architectures": [
4
- "ViTForImageClassification"
5
  ],
6
  "model_type": "vit",
7
  "num_labels": 2,
 
1
  {
2
  "_name_or_path": "jacoballessio/ai-image-detect-distilled",
3
  "architectures": [
4
+ "efficientnet"
5
  ],
6
  "model_type": "vit",
7
  "num_labels": 2,