Update README.md
Browse files
README.md
CHANGED
|
@@ -100,7 +100,7 @@ tokenized_input = tokenizer(input, add_special_tokens= False)
|
|
| 100 |
|
| 101 |
Using model with Jax and Pytorch
|
| 102 |
```python
|
| 103 |
-
from transformers import AutoTokenizer, FlaxAutoModelForSequenceClassification
|
| 104 |
|
| 105 |
#Load jax model
|
| 106 |
model = FlaxAutoModelForSequenceClassification.from_pretrained("Fsoft-AIC/Codebert-docstring-inconsistency")
|
|
|
|
| 100 |
|
| 101 |
Using model with Jax and Pytorch
|
| 102 |
```python
|
| 103 |
+
from transformers import AutoTokenizer, AutoModelForSequenceClassification, FlaxAutoModelForSequenceClassification
|
| 104 |
|
| 105 |
#Load jax model
|
| 106 |
model = FlaxAutoModelForSequenceClassification.from_pretrained("Fsoft-AIC/Codebert-docstring-inconsistency")
|