MPA commited on
Commit
fa7f994
verified
1 Parent(s): f12b1ab

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -57,8 +57,8 @@ def mean_pooling(model_output, attention_mask):
57
  sentences = ["讗诪讗 讛诇讻讛 诇讙谉", "讗讘讗 讛诇讱 诇讙谉", "讬专拽讜谞讬 拽讜谞讛 诇谞讜 驻讬爪讜转"]
58
 
59
  # Load model from HuggingFace Hub
60
- tokenizer = AutoTokenizer.from_pretrained('{MODEL_NAME}')
61
- model = AutoModel.from_pretrained('{MODEL_NAME}')
62
 
63
  # Tokenize sentences
64
  encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')
 
57
  sentences = ["讗诪讗 讛诇讻讛 诇讙谉", "讗讘讗 讛诇讱 诇讙谉", "讬专拽讜谞讬 拽讜谞讛 诇谞讜 驻讬爪讜转"]
58
 
59
  # Load model from HuggingFace Hub
60
+ tokenizer = AutoTokenizer.from_pretrained('MPA/sambert')
61
+ model = AutoModel.from_pretrained('MPA/sambert')
62
 
63
  # Tokenize sentences
64
  encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')