Spaces:
Sleeping
Sleeping
Commit
•
c1f7fcb
1
Parent(s):
5a927fa
update embedding path
Browse files
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(...),
|