Create handler.py

handler.py

@@ -0,0 +1,12 @@
+from transformers import AutoTokenizer, pipeline
+from optimum.onnxruntime import ORTModelForSequenceClassification
+
+model_id = "."
+
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = ORTModelForSequenceClassification.from_pretrained(model_id)
+
+classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
+
+def __call__(data):
+    return classifier(data["inputs"])