diff --git a/README.md b/README.md index 3868ed20..6f9e1fe3 100644 --- a/README.md +++ b/README.md @@ -50,10 +50,17 @@ embeddings = np.array([data.embedding for data in data]) ### To use multimodal embedding: we support CLIP +### Requirements Directory with pictures you want to search ```python -data:list[EmbedData] = embed_anything.embed_file("filename.pdf", embeder= "Clip") +data:list[EmbedData] = embed_anything.embed_directory("test_files", embeder= "Clip") embeddings = np.array([data.embedding for data in data]) + +query = "photo of a dog" +query_embedding = np.array(embed_anything.embed_query(query, embeder= "Clip")[0].embedding) +similarities = np.dot(embeddings, query_embedding) +max_index = np.argmax(similarities) +Image.open(data[max_index].text).show() ```