update README.md

LarFii · LarFii · commit 7409668f9f27 · 2024-10-15T21:23:03.000+08:00
diff --git 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
