Update README.md

README.md

@@ -36,36 +36,12 @@ image = Image.open(requests.get(url, stream=True).raw)
 processor = AutoImageProcessor.from_pretrained(
     "apple/aimv2-large-patch14-native",
 )
-model = AutoModel.from_pretrained(
-    "apple/aimv2-large-patch14-native",
-    trust_remote_code=True,
-)
+model = AutoModel.from_pretrained("apple/aimv2-large-patch14-native")
 
 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-native",
-)
-model = FlaxAutoModel.from_pretrained(
-    "apple/aimv2-large-patch14-native",
-    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