zeroshot
/

sst2-distilbert-sparse

Text Classification

Model card Files Files and versions Community

zeroshot commited on Nov 14, 2022

Commit

c59edad

·

1 Parent(s): 7c4cbdd

edit handler

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -4,7 +4,7 @@ from time import perf_counter
 class EndpointHandler:
-    def __init__(self, path="deployment/model.onnx"):
         self.pipeline = Pipeline.create(task="text-classification", model_path=path)

 class EndpointHandler:
+    def __init__(self, path="./deployment/model.onnx"):
         self.pipeline = Pipeline.create(task="text-classification", model_path=path)