Ahmed Tarek commited on
Commit
fe7ebc4
·
1 Parent(s): 7b5c9ca
services/embedding_models/MiniLM_L12_v2_model.py CHANGED
@@ -11,7 +11,7 @@ logging.set_verbosity_error()
11
 
12
  class ONNXMiniLMModel:
13
  def __init__(self,
14
- model_name="old_models/paraphrase-multilingual-MiniLM-L12-v2/0_Transformer",
15
  onnx_path="/tmp/onnx_model/minilm.onnx"): # Different ONNX path
16
 
17
  self.model_name = model_name
 
11
 
12
  class ONNXMiniLMModel:
13
  def __init__(self,
14
+ model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
15
  onnx_path="/tmp/onnx_model/minilm.onnx"): # Different ONNX path
16
 
17
  self.model_name = model_name
services/embedding_models/zrt.py ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ from transformers import AutoTokenizer
2
+ tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")