Hugging Face's logo

tbs17
/
MathBERT-custom

Fill-Mask
Transformers
PyTorch
bert
Inference Endpoints
Model card Files Community
2

Pytorch example encoded_input fix

#1
by DimOgu - opened
base: refs/heads/main
←
from: refs/pr/1
Discussion Files changed
+1
-1
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -49,7 +49,7 @@ Here is how to use this model to get the features of a given text in PyTorch:
49
  tokenizer = BertTokenizer.from_pretrained('tbs17/MathBERT-custom')
50
  model = BertModel.from_pretrained("tbs17/MathBERT-custom")
51
  text = "Replace me by any text you'd like."
52
- encoded_input = tokenizer(text, return_tensors='pt')
53
  output = model(encoded_input)
54
  ```
55
  and in TensorFlow:
 
49
  tokenizer = BertTokenizer.from_pretrained('tbs17/MathBERT-custom')
50
  model = BertModel.from_pretrained("tbs17/MathBERT-custom")
51
  text = "Replace me by any text you'd like."
52
+ encoded_input = tokenizer(text, return_tensors='pt')["input_ids"]
53
  output = model(encoded_input)
54
  ```
55
  and in TensorFlow: