From afc5f0217034c3d915ade49f4998f4ea33e23f26 Mon Sep 17 00:00:00 2001
From: wxywb <xy.wang@zilliz.com>
Date: Fri, 10 Feb 2023 09:36:03 +0000
Subject: [PATCH] update the operator.

Signed-off-by: wxywb <xy.wang@zilliz.com>
---
 README.md | 2 +-
 blip.py   | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

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