small

app.py

@@ -19,7 +19,7 @@ def decrypt_file(input_path, key):
     return plaintext.decode('utf-8')
 
 
-llm = llama_cpp.Llama.from_pretrained(repo_id="KimChen/bge-m3-GGUF", filename="bge-m3-q4_k_m.gguf", embedding=True)
+llm = llama_cpp.Llama.from_pretrained(repo_id="mradermacher/bge-large-zh-v1.5-GGUF", filename="bge-large-zh-v1.5.Q4_K_M.gguf", embedding=True)
 
 embedding_1 = llm.create_embedding("Hello, world!")
 embedding_2 = llm.create_embedding("你好, 世界!") # type(embedding_1['data'][0]['embedding']) list
@@ -40,7 +40,7 @@ raw_jsons = json.loads(decrypted_content)
 docs = []
 metas = []
 
-for vhjx_item in raw_jsons:
+for vhjx_item in raw_jsons[:1]:
     chapter = vhjx_item[0]
     for jvvi_item in vhjx_item[1:]:
         content = jvvi_item["原文"]