from search_engine_model import SearchEngineModel | |
def main(): | |
search_engine_model = SearchEngineModel() | |
model, preprocess = search_engine_model.load_clip_model() | |
input_image_path = './cat.jpg' | |
image_embeddings = search_engine_model.encode_image(model, preprocess, input_image_path) | |
main() |