Update README.md

README.md

@@ -79,15 +79,13 @@ Users should evaluate the model on their specific crops and farming conditions.
 ## How to Get Started with the Model
 
 ```python
-import matplotlib.pyplot as plt
 from PIL import Image, UnidentifiedImageError
 from transformers import ViTFeatureExtractor, ViTForImageClassification
-feature_extractor = ViTFeatureExtractor.from_pretrained('wambugu1738/crop_leaf_diseases_vit')
+feature_extractor = ViTFeatureExtractor.from_pretrained('wambugu71/crop_leaf_diseases_vit')
 model = ViTForImageClassification.from_pretrained(
     'wambugu1738/crop_leaf_diseases_vit',
     ignore_mismatched_sizes=True
 )
-from PIL import Image
 image = Image.open('<image_path>')
 inputs = feature_extractor(images=image, return_tensors="pt")
 outputs = model(**inputs)