diff --git a/README.md b/README.md
index af94275..ad1d3f8 100644
--- a/README.md
+++ b/README.md
@@ -39,7 +39,7 @@ text_pipe = (
     .output('text', 'vec')
 )
 
-DataCollection(image_pipe('./teddy.jpg')).show()
+DataCollection(img_pipe('./teddy.jpg')).show()
 DataCollection(text_pipe('A teddybear on a skateboard in Times Square.')).show()
 
 ```
diff --git a/blip.py b/blip.py
index b17c577..1c04ca6 100644
--- a/blip.py
+++ b/blip.py
@@ -13,9 +13,12 @@
 # limitations under the License.
 
 import sys
+import os
 from pathlib import Path
 
 import torch
+import logging
+import warnings
 from torch import nn
 
 from transformers import AutoProcessor, BlipForImageTextRetrieval