Update README.md

README.md

@@ -89,6 +89,8 @@ model = AutoModelForSequenceClassification.from_pretrained(
       trust_remote_code=False,
       ignore_mismatched_sizes=False,
 )
+model.cuda()
+model.eval()
 tokenizer = AutoTokenizer.from_pretrained(
     model_dir,
     use_fast=True,