Pranjal12345
/

Classification_Transformers

Image Classification

ImageClassification

Model card Files Files and versions

Pranjal12345 commited on Sep 19, 2023

Commit

756b621

·

1 Parent(s): d3971ae

rtr

Files changed (1) hide show

config.json +9 -1

config.json CHANGED Viewed

@@ -7,5 +7,13 @@
     "4": "engaged",
     "5": "frustated"
   },
-  "model_type": "vit_b_16"
 }

     "4": "engaged",
     "5": "frustated"
   },
+  "model_type": "vit",
+  "_name_or_path": [
+    "google/vit-base-patch16-224-in21k"
+  ],
+  "architectures": [
+    [
+      "ViTForImageClassification"
+    ]
+  ]
 }