From 7409668f9f2726c9f702a5ac9ed01e415e9b97b3 Mon Sep 17 00:00:00 2001
From: LarFii <834462287@qq.com>
Date: Tue, 15 Oct 2024 21:23:03 +0800
Subject: [PATCH] update README.md

---
 README.md | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index cc282c27..12a422f6 100644
--- a/README.md
+++ b/README.md
@@ -86,9 +86,16 @@ rag = LightRAG(
     working_dir=WORKING_DIR,
     llm_model_func=hf_model_complete,  # Use Hugging Face complete model for text generation
     llm_model_name='meta-llama/Llama-3.1-8B-Instruct',  # Model name from Hugging Face
-    embedding_func=hf_embedding,  # Use Hugging Face embedding function
-    tokenizer=AutoTokenizer.from_pretrained("sentence-transformers/all-MiniLM-L6-v2"),
-    embed_model=AutoModel.from_pretrained("sentence-transformers/all-MiniLM-L6-v2")
+    # Use Hugging Face embedding function
+    embedding_func=EmbeddingFunc(
+        embedding_dim=384,
+        max_token_size=5000,
+        func=lambda texts: hf_embedding(
+            texts, 
+            tokenizer=AutoTokenizer.from_pretrained("sentence-transformers/all-MiniLM-L6-v2"),
+            embed_model=AutoModel.from_pretrained("sentence-transformers/all-MiniLM-L6-v2")
+        )
+    ),
 )
 ```
 ### Batch Insert