test with another model

handler.py

@@ -1,13 +1,22 @@
 from typing import  Dict, List, Any
-from optimum.onnxruntime import ORTModelForSequenceClassification
+# from optimum.onnxruntime import ORTModelForSequenceClassification
+from transformers import AutoModel
+
 from transformers import pipeline, AutoTokenizer
 
+checkpoint = "distilbert-base-uncased-finetuned-sst-2-english"
 
 class EndpointHandler():
+    
     def __init__(self, path=""):
         # load the optimized model
-        model = ORTModelForSequenceClassification.from_pretrained(path)
-        tokenizer = AutoTokenizer.from_pretrained(path)
+        # model = ORTModelForSequenceClassification.from_pretrained(path)
+        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)
+
         # create inference pipeline
         self.pipeline = pipeline("text-classification", model=model, tokenizer=tokenizer)