YaYaB
/

yb_test_inference_endpoint_text_cls

Model card Files Files and versions Community

YassineB commited on Oct 17, 2022

Commit

423dcb9

·

1 Parent(s): 983abe6

Fix requirements

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -4,7 +4,7 @@ from transformers import pipeline, AutoTokenizer
 class EndpointHandler():
-    def __init__(self):
         # load the optimized model
         path = "optimum/roberta-base-squad2"
         model = ORTModelForSequenceClassification.from_pretrained(path)

 class EndpointHandler():
+    def __init__(self, path=""):
         # load the optimized model
         path = "optimum/roberta-base-squad2"
         model = ORTModelForSequenceClassification.from_pretrained(path)