Update calc_cosine_similarity.py

calc_cosine_similarity.py

@@ -20,8 +20,14 @@ def find_cosine_similarity(text1, text2):
 def find_embedding(texts):
     global model
     embeddings = []
-    
+
+    c = 0
+    lim = 10
     for text in texts:
+        if c > lim:
+            continue
+        else:
+            c += 1
         print(f"Finding embedding for {text}")
         embeddings.append(model.encode(text, convert_to_tensor=True))