testfile added

test.py

@@ -4,29 +4,22 @@
 from PIL import Image
 import time
 
-# start = time.time()
-# data= embed_anything.embed_file("test_files/clip/cat1.jpg", embeder= "Clip")
+start = time.time()
+data= embed_anything.embed_file("test_files/clip/cat1.jpg", embeder= "Clip")
 
 
-# embeddings = np.array([data.embedding for data in data])
+embeddings = np.array([data.embedding for data in data])
 
-# print(data[0])
+print(data[0])
 
-# query = ["Photo of a dog?"]
-# query_embedding = np.array(embed_anything.embed_query(query, embeder= "Clip")[0].embedding)
+query = ["Photo of a dog?"]
+query_embedding = np.array(embed_anything.embed_query(query, embeder= "Clip")[0].embedding)
 
-# similarities = np.dot(embeddings, query_embedding)
+similarities = np.dot(embeddings, query_embedding)
 
-# max_index = np.argmax(similarities)
+max_index = np.argmax(similarities)
 
-# # Image.open(data[max_index].text).show()
-# print(data[max_index].text)
-# end = time.time()
-# print("Time taken: ", end-start)
-
-
-url = "https://www.akshaymakes.com/blogs/3d_convolution"
-
-data = embed_anything.emb_webpage(url, embeder= "Bert")
-
-print(data[0])
+# Image.open(data[max_index].text).show()
+print(data[max_index].text)
+end = time.time()
+print("Time taken: ", end-start)