Update README.md

README.md

@@ -25,11 +25,13 @@ def resize_with_padding(image, target_size=(224, 224)):
     ])
     return transform((padded_img))
 
-path="0106.jpg"
-img=resize_with_padding(Image.open(path))
+path="/home/t2/Downloads/Figure_4.png"
+img=resize_with_padding(Image.open(path).convert('RGB'))
 model.eval()
 with torch.no_grad():
-    print(tokenizer.batch_decode(model.cuda().generate(img.unsqueeze(0).cuda()),skip_special_tokens=True))
+    ans=tokenizer.batch_decode(model.cuda().generate(img.unsqueeze(0).cuda()),skip_special_tokens=True)
+    print(ans)
 
 plt.imshow(img.permute(1,2,0).detach().cpu().numpy())
+plt.show()
 ```