rapminerz
/

WatiBERT-large-cased

Inference Endpoints

Model card Files Files and versions Community

RapMinerz commited on Jul 18

Commit

2527cf8

•

1 Parent(s): 0f62846

update readme

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -53,11 +53,11 @@ pip install transformers
 To load the WatiBERT model, use the following Python code:
 ```python
-from transformers import AutoModel, AutoTokenizer
 # Load the tokenizer and model
-tokenizer = AutoTokenizer.from_pretrained("rapminerz/WatiBERT-large-cased")
-model = AutoModel.from_pretrained("rapminerz/WatiBERT-large-cased")
 ```
 ## Using the Model

 To load the WatiBERT model, use the following Python code:
 ```python
+from transformers import FlaubertTokenizer, FlaubertWithLMHeadModel
 # Load the tokenizer and model
+tokenizer = FlaubertTokenizer.from_pretrained("rapminerz/WatiBERT-large-cased")
+model = FlaubertWithLMHeadModel.from_pretrained("rapminerz/WatiBERT-large-cased")
 ```
 ## Using the Model