jacob-braun-mn commited on
Commit
c1f7fcb
1 Parent(s): 5a927fa

update embedding path

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -19,7 +19,7 @@ app.add_middleware(
19
  )
20
 
21
  model_pipe = pipeline("text-generation", model="./models/Qwen2-1dot5B-Instruct")
22
- embeddings = HuggingFaceEmbeddings(model_name="all-mpnet-base-v2")
23
 
24
  @app.post("/search_document")
25
  async def retrieve_record(textFile: UploadFile = File(...),
 
19
  )
20
 
21
  model_pipe = pipeline("text-generation", model="./models/Qwen2-1dot5B-Instruct")
22
+ embeddings = HuggingFaceEmbeddings(model_name="./models/all-mpnet-base-v2")
23
 
24
  @app.post("/search_document")
25
  async def retrieve_record(textFile: UploadFile = File(...),