Lambuda commited on
Commit
9710db9
·
1 Parent(s): c5c69a3
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -17,7 +17,7 @@ async def process_post_request(request: Request):
17
  # 在这里可以对接收到的数据进行处理
18
  # 例如,可以通过data["key"]来访问请求体中的参数
19
  print(data)
20
- embeddings = HuggingFaceEmbeddings(model_name="GanymedeNil/text2vec-large-chinese", model_kwargs={'device': "cpu","cache_dir":'/cache'})
21
  result= embeddings.embed_query(data['data'])
22
  return result
23
 
 
17
  # 在这里可以对接收到的数据进行处理
18
  # 例如,可以通过data["key"]来访问请求体中的参数
19
  print(data)
20
+ embeddings = HuggingFaceEmbeddings(model_name="GanymedeNil/text2vec-large-chinese",model_kwargs={'device': "cpu"},cache_folder="/cache")
21
  result= embeddings.embed_query(data['data'])
22
  return result
23