Update README.md

README.md

@@ -98,17 +98,12 @@ tokenized_input = tokenizer(input, add_special_tokens= False)
 """
 ```
 
-Using model with Jax
+Using model with Jax and Pytorch
 ```python
 from transformers import AutoTokenizer, FlaxAutoModelForSequenceClassification
 
 #Load jax model
 model = FlaxAutoModelForSequenceClassification.from_pretrained("Fsoft-AIC/Codebert-docstring-inconsistency")
-```
-
-Using model with Pytorch
-```python
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 #Load torch model
 model = AutoModelForSequenceClassification.from_pretrained("Fsoft-AIC/Codebert-docstring-inconsistency")