use minilm-l12 embedding model
Browse files
main.py
CHANGED
|
@@ -23,7 +23,7 @@ app.add_middleware(
|
|
| 23 |
allow_headers=["*"],
|
| 24 |
)
|
| 25 |
|
| 26 |
-
embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/paraphrase-MiniLM-
|
| 27 |
|
| 28 |
@app.post("/uploadfile/")
|
| 29 |
async def upload_file(file: UploadFile = File(...)):
|
|
@@ -56,6 +56,4 @@ async def query(request: QueryRequest):
|
|
| 56 |
return {"question": request.question, "answer": answer}
|
| 57 |
except Exception as e:
|
| 58 |
raise HTTPException(status_code=500, detail=str(e))
|
| 59 |
-
|
| 60 |
-
import uvicorn
|
| 61 |
-
uvicorn.run(app, host="0.0.0.0", port=int(os.getenv("PORT", 8000)))
|
|
|
|
| 23 |
allow_headers=["*"],
|
| 24 |
)
|
| 25 |
|
| 26 |
+
embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/paraphrase-MiniLM-L12-v2")
|
| 27 |
|
| 28 |
@app.post("/uploadfile/")
|
| 29 |
async def upload_file(file: UploadFile = File(...)):
|
|
|
|
| 56 |
return {"question": request.question, "answer": answer}
|
| 57 |
except Exception as e:
|
| 58 |
raise HTTPException(status_code=500, detail=str(e))
|
| 59 |
+
|
|
|
|
|
|