Update README.md

README.md

@@ -117,10 +117,16 @@ try:
         "Deep learning uses neural networks with multiple layers to process data.",
         "Neural networks are computing systems inspired by biological neural networks.",
         "Artificial intelligence encompasses machine learning and deep learning.",
-        "Here is how you train dogs",
     ]
+
+    # Encode and find similarity
+    q_reps = colbert.encode_queries(queries, batch_size=4, to_cpu=True)
+    p_reps = colbert.encode_documents(documents, batch_size=4, to_cpu=True)
+        
+    print("Computing similarities...")
+    scores = colbert.compute_similarity(q_reps, p_reps)
     
-    # Test single query ranking
+    # or Test single query ranking
     print("\n=== Single Query Ranking ===")
     query = "How does deep learning work?"
     results = colbert.rank_documents(query, documents, top_k=3)
@@ -501,10 +507,16 @@ documents = [
     "Deep learning uses neural networks with multiple layers to process data.",
     "Neural networks are computing systems inspired by biological neural networks.",
     "Artificial intelligence encompasses machine learning and deep learning.",
-    "Here is how you train dogs",
 ]
 
-# Test single query ranking
+# Encode and find similarity
+q_reps = onnx_colbert.encode_queries(queries, batch_size=4, to_cpu=True)
+p_reps = onnx_colbert.encode_documents(documents, batch_size=4, to_cpu=True)
+print("Computing similarities...")
+scores = onnx_colbert.compute_similarity(q_reps, p_reps)
+
+
+# or Test single query ranking
 print("\n=== ONNX Standalone Single Query Ranking ===")
 query = "How does deep learning work?"
 results = onnx_colbert.rank_documents(query, documents, top_k=3)