Update README.md

README.md

@@ -16,7 +16,7 @@ widget:
 
 # LeViT
 
-LeViT128S model pre-trained on ImageNet-1k at resolution 224x224. It was introduced in the paper [LeViT: a Vision Transformer in ConvNet's Clothing for Faster Inference
+LeViT-256 model pre-trained on ImageNet-1k at resolution 224x224. It was introduced in the paper [LeViT: a Vision Transformer in ConvNet's Clothing for Faster Inference
 ](https://arxiv.org/abs/2104.01136) by Graham et al. and first released in [this repository](https://github.com/facebookresearch/LeViT). 
 
 Disclaimer: The team releasing LeViT did not write a model card for this model so this model card has been written by the Hugging Face team.
@@ -33,8 +33,8 @@ import requests
 url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
 image = Image.open(requests.get(url, stream=True).raw)
 
-feature_extractor = LevitFeatureExtractor.from_pretrained('anugunj/levit-256')
-model = LevitForImageClassificationWithTeacher.from_pretrained('anugunj/levit-256')
+feature_extractor = LevitFeatureExtractor.from_pretrained('facebook/levit-256')
+model = LevitForImageClassificationWithTeacher.from_pretrained('facebook/levit-256')
 
 inputs = feature_extractor(images=image, return_tensors="pt")
 outputs = model(**inputs)