Thytu commited on
Commit
d37e863
1 Parent(s): 097107d

Update bhadresh-savani_distilbert-base-uncased-emotion_onnx_tokenize/1/model.py

Browse files
bhadresh-savani_distilbert-base-uncased-emotion_onnx_tokenize/1/model.py CHANGED
@@ -42,7 +42,8 @@ class TritonPythonModel:
42
  """
43
  # more variables in https://github.com/triton-inference-server/python_backend/blob/main/src/python.cc
44
 
45
- path: str = str(Path(args["model_repository"]).parent.absolute())
 
46
  self.tokenizer = AutoTokenizer.from_pretrained(path)
47
 
48
  def execute(self, requests) -> "List[List[pb_utils.Tensor]]":
 
42
  """
43
  # more variables in https://github.com/triton-inference-server/python_backend/blob/main/src/python.cc
44
 
45
+ path: str = str(Path(args["model_repository"]).absolute()) + "/1/"
46
+
47
  self.tokenizer = AutoTokenizer.from_pretrained(path)
48
 
49
  def execute(self, requests) -> "List[List[pb_utils.Tensor]]":