| import numpy as np | |
| import pandas as pd | |
| from search_engine_model import SearchEngineModel | |
| def main(): | |
| search_engine_model = SearchEngineModel() | |
| model, preprocess = search_engine_model.load_clip_model() | |
| text_prompt = 'cat' | |
| text_embeddings = search_engine_model.encode_text(model, text_prompt) | |
| random_features = np.random.rand(50, 512) | |
| input_df = pd.DataFrame(random_features) | |
| names_column = [f'image_{it}.png' for it in range(0, len(random_features))] | |
| input_df.insert(0, 'images_names', names_column) | |
| search_result = search_engine_model.search_image_by_text_prompt(text_embeddings, input_df) | |
| main() |