fix config and readme

README.md

@@ -23,7 +23,7 @@ Uses `NOT-A-QUESTION`, `FACTOID`, `DEBATE`, `EVIDENCE-BASED`, `INSTRUCTION`, `RE
 
 ## Citation
 
-If you use `NDQA-cats` in your work, please cite [this paper](https://dl.acm.org/doi/10.1145/3477495.3531926)
+If you use `NFQA-cats` in your work, please cite [this paper](https://dl.acm.org/doi/10.1145/3477495.3531926)
 
 ```
 @misc{bolotova2022nfcats,

config.json

@@ -4,7 +4,10 @@
   ],
   "auto_map": {
     "AutoConfig": "RobertaConfig",
-    "AutoModelForImageClassification": "nfqa_model.RobertaNFQAClassification"
+    "AutoModelForSequenceClassification": {
+      "pretrained_model_name_or_path": "nfqa_model.RobertaNFQAClassification",
+      "trust_remote_code": true
+    }
   },
   "attention_probs_dropout_prob": 0.1,
   "bos_token_id": 0,