Update README.md

README.md

@@ -1,5 +1,8 @@
 ```python
+from PIL import Image
+import matplotlib.pyplot as plt
 from transformers import VisionEncoderDecoderModel, AutoTokenizer
+
 tokenizer = AutoTokenizer.from_pretrained('tirthadagr8/CustomOCR')
 model=VisionEncoderDecoderModel.from_pretrained('tirthadagr8/CustomOCR')
 import torch
@@ -9,12 +12,12 @@ simple_transforms=T.Compose([
             T.ToTensor(),
             T.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
         ])
-from PIL import Image
+
 path="image.jpg"
 img=simple_transforms(Image.open(path))
 model.eval()
 with torch.no_grad():
     print(tokenizer.batch_decode(model.cuda().generate(img.unsqueeze(0).cuda()),skip_special_tokens=True))
-import matplotlib.pyplot as plt
+
 plt.imshow(Image.open(path).resize((224,224)))
 ```