priyabrat
/

UpTag_GenderClassification

Text Classification

Model card Files Files and versions

priyabrat commited on Jan 16, 2023

Commit

db0e865

·

1 Parent(s): 3c39b73

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -2,7 +2,7 @@
   "_name_or_path": "priyabrat/UpTag_GenderClassification",
   "activation": "gelu",
   "architectures": [
-    "DistilBertForSequenceClassification"
   ],
   "attention_dropout": 0.1,
   "dim": 768,

   "_name_or_path": "priyabrat/UpTag_GenderClassification",
   "activation": "gelu",
   "architectures": [
+    "AutoModelForSequenceClassification"
   ],
   "attention_dropout": 0.1,
   "dim": 768,