initial

main.py

@@ -1,6 +1,7 @@
 
 from fastapi import FastAPI,Request
 from langchain.embeddings import HuggingFaceEmbeddings
+from sentence_transformers import SentenceTransformer
 
 app = FastAPI()
 
@@ -16,13 +17,13 @@ async def process_post_request(request: Request):
     # 在这里可以对接收到的数据进行处理
     # 例如，可以通过data["key"]来访问请求体中的参数
     print(data)
-    embeddings = HuggingFaceEmbeddings(model_name="GanymedeNil/text2vec-large-chinese", model_kwargs={'device': "cpu"})
+    embeddings = HuggingFaceEmbeddings(model_name="GanymedeNil/text2vec-large-chinese", model_kwargs={'device': "cpu","cache_dir":'/cache'})
     result= embeddings.embed_query(data['data'])
     return result
 
 
 if __name__ == '__main__':
-    embeddings = HuggingFaceEmbeddings(model_name="M:\\hugging face model\\text2vec-large-chinese", model_kwargs={'device': "cpu"})
+    embeddings = HuggingFaceEmbeddings(model_name="GanymedeNil/text2vec-large-chinese", model_kwargs={'device': "cpu"})
     result= embeddings.embed_query("你好")
     print(result)