Blaxzter
/

LaBSE-sentence-embeddings

Sentence Similarity

feature-extraction

sentence_embedding

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

Blaxzter commited on May 3, 2023

Commit

a38b233

·

1 Parent(s): 1569966

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -5,7 +5,7 @@ from transformers import BertModel, BertTokenizerFast
 class EndpointHandler():
-    def __init__(self, path_to_model: str):
         # Preload all the elements you are going to need at inference.
         # pseudo:
         self.tokenizer = BertTokenizerFast.from_pretrained(path_to_model)

 class EndpointHandler():
+    def __init__(self, path_to_model: str = '.'):
         # Preload all the elements you are going to need at inference.
         # pseudo:
         self.tokenizer = BertTokenizerFast.from_pretrained(path_to_model)