File size: 636 Bytes
08614a1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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() |