Update README.md

README.md

@@ -38,20 +38,8 @@ We release the scripts to evaluate our model's performance [here](https://github
 from sentence_transformers import SentenceTransformer
 
 model = SentenceTransformer("cornstack/CodeRankEmbed", trust_remote_code=True)
-queries = ['Represent this query for searching relevant code: Calculate the n-th Fibonacci number']
-codes = ["""def func(n):
-    if n <= 0:
-        return "Input should be a positive integer"
-    elif n == 1:
-        return 0
-    elif n == 2:
-        return 1
-    else:
-        a, b = 0, 1
-        for _ in range(2, n):
-            a, b = b, a + b
-        return b
-"""]
+queries = ['Represent this query for searching relevant code: Calculate the n-th factorial']
+codes = ['def fact(n):\n if n < 0:\n  raise ValueError\n return 1 if n == 0 else n * fact(n - 1)']
 query_embeddings = model.encode(queries)
 print(query_embeddings)
 code_embeddings = model.encode(codes)