test with seqclass

handler.py

@@ -1,6 +1,7 @@
 from typing import  Dict, List, Any
 # from optimum.onnxruntime import ORTModelForSequenceClassification
-from transformers import AutoModel
+# from transformers import AutoModel
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
 from transformers import pipeline, AutoTokenizer
 
@@ -11,8 +12,8 @@ class EndpointHandler():
     def __init__(self, path=""):
         # load the optimized model
         # model = ORTModelForSequenceClassification.from_pretrained(path)
-        model = AutoModel.from_pretrained(checkpoint)
-        # model = AutoModelForSequenceClassification.from_pretrained(checkpoint)
+        # model = AutoModel.from_pretrained(checkpoint)
+        model = AutoModelForSequenceClassification.from_pretrained(checkpoint)
 
         # tokenizer = AutoTokenizer.from_pretrained(path)
         tokenizer = AutoTokenizer.from_pretrained(pretrained_model_name_or_path=checkpoint)