Update README.md

README.md

@@ -195,34 +195,12 @@ processor = AutoImageProcessor.from_pretrained(
 )
 model = AutoModel.from_pretrained(
     "apple/aimv2-large-patch14-224",
-    trust_remote_code=True,
 )
 
 inputs = processor(images=image, return_tensors="pt")
 outputs = model(**inputs)
 ```
 
-### JAX
-```python
-import requests
-from PIL import Image
-from transformers import AutoImageProcessor, FlaxAutoModel
-
-url = "http://images.cocodataset.org/val2017/000000039769.jpg"
-image = Image.open(requests.get(url, stream=True).raw)
-
-processor = AutoImageProcessor.from_pretrained(
-    "apple/aimv2-large-patch14-224",
-)
-model = FlaxAutoModel.from_pretrained(
-    "apple/aimv2-large-patch14-224",
-    trust_remote_code=True,
-)
-
-inputs = processor(images=image, return_tensors="jax")
-outputs = model(**inputs)
-```
-
 ## Citation
 If you find our work useful, please consider citing us as:
 ```bibtex