test wo seqclass

handler.py

@@ -1,7 +1,7 @@
 from typing import  Dict, List, Any
 # from optimum.onnxruntime import ORTModelForSequenceClassification
-# from transformers import AutoModel
-from transformers import AutoModelForSequenceClassification, AutoTokenizer
+from transformers import AutoModel
+# from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
 from transformers import pipeline, AutoTokenizer
 
@@ -12,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)