Hugging Face's logo

zeroshot
/
sst2-distilbert-sparse

Text Classification
Transformers
ONNX
distilbert
Model card Files Community
zeroshot commited on
Commit
7c4cbdd
·
1 Parent(s): e0d8ae6

edit handler

Browse files
Files changed (5) hide show
  1. config.json → deployment/config.json +0 -0
  2. model.onnx → deployment/model.onnx +0 -0
  3. tokenizer.json → deployment/tokenizer.json +0 -0
  4. tokenizer_config.json → deployment/tokenizer_config.json +0 -0
  5. handler.py +1 -1
config.json → deployment/config.json RENAMED
File without changes
model.onnx → deployment/model.onnx RENAMED
File without changes
tokenizer.json → deployment/tokenizer.json RENAMED
File without changes
tokenizer_config.json → deployment/tokenizer_config.json RENAMED
File without changes
handler.py CHANGED
@@ -4,7 +4,7 @@ from time import perf_counter
4
 
5
  class EndpointHandler:
6
 
7
- def __init__(self, path=""):
8
 
9
  self.pipeline = Pipeline.create(task="text-classification", model_path=path)
10
 
 
4
 
5
  class EndpointHandler:
6
 
7
+ def __init__(self, path="deployment/model.onnx"):
8
 
9
  self.pipeline = Pipeline.create(task="text-classification", model_path=path)
10