klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added Clip dependency (#778)

Browse Source
This commit is contained in:
Rupesh Bansal
2023-10-10 00:32:45 +05:30
committed by GitHub
parent bc649b9a85
commit 19a9141c2d
4 changed files with 47 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
embedchain/loaders/images.py
View File

@@ -16,15 +16,15 @@ class ImagesLoader(BaseLoader):
# load model and image preprocessing
from embedchain.models.clip_processor import ClipProcessor
model, preprocess = ClipProcessor.load_model()
model = ClipProcessor.load_model()
if os.path.isfile(image_url):
data = [ClipProcessor.get_image_features(image_url, model, preprocess)]
data = [ClipProcessor.get_image_features(image_url, model)]
else:
data = []
for filename in os.listdir(image_url):
filepath = os.path.join(image_url, filename)
try:
data.append(ClipProcessor.get_image_features(filepath, model, preprocess))
data.append(ClipProcessor.get_image_features(filepath, model))
except Exception as e:
# Log the file that was not loaded
logging.exception("Failed to load the file {}. Exception {}".format(filepath, e))