tbs17
/

MathBERT-custom

Inference Endpoints

Model card Files Files and versions Community

tbs17 commited on Jul 4, 2022

Commit

632c8c4

•

1 Parent(s): 92c41fd

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -49,7 +49,7 @@ Here is how to use this model to get the features of a given text in PyTorch:
 tokenizer = BertTokenizer.from_pretrained('tbs17/MathBERT-custom')
 model = BertModel.from_pretrained("tbs17/MathBERT-custom")
 text = "Replace me by any text you'd like."
-encoded_input = tokenizer(text, return_tensors='pt')['input_ids']
 output = model(encoded_input)
 ```
 and in TensorFlow:

 tokenizer = BertTokenizer.from_pretrained('tbs17/MathBERT-custom')
 model = BertModel.from_pretrained("tbs17/MathBERT-custom")
 text = "Replace me by any text you'd like."
+encoded_input = tokenizer(text, return_tensors='pt')
 output = model(encoded_input)
 ```
 and in TensorFlow: