Spaces:
Sleeping
Sleeping
Commit
·
aaec9ae
1
Parent(s):
0623b01
add startup logs
Browse files
server.py
CHANGED
|
@@ -42,14 +42,18 @@ class ML(BaseModel):
|
|
| 42 |
|
| 43 |
|
| 44 |
def load_models(app: FastAPI) -> FastAPI:
|
|
|
|
| 45 |
retriever=BGEM3FlagModel('BAAI/bge-m3', use_fp16=True) ,
|
|
|
|
| 46 |
ranker=FlagReranker('BAAI/bge-reranker-v2-m3', use_fp16=True)
|
|
|
|
| 47 |
ml = ML(
|
| 48 |
retriever=retriever,
|
| 49 |
ranker=ranker,
|
| 50 |
data=get_data(retriever)
|
| 51 |
)
|
| 52 |
app.ml = ml
|
|
|
|
| 53 |
return app
|
| 54 |
|
| 55 |
|
|
|
|
| 42 |
|
| 43 |
|
| 44 |
def load_models(app: FastAPI) -> FastAPI:
|
| 45 |
+
logger.info("Loading embedding model...")
|
| 46 |
retriever=BGEM3FlagModel('BAAI/bge-m3', use_fp16=True) ,
|
| 47 |
+
logger.info("Loading ranker model...")
|
| 48 |
ranker=FlagReranker('BAAI/bge-reranker-v2-m3', use_fp16=True)
|
| 49 |
+
logger.info("Done loading models!")
|
| 50 |
ml = ML(
|
| 51 |
retriever=retriever,
|
| 52 |
ranker=ranker,
|
| 53 |
data=get_data(retriever)
|
| 54 |
)
|
| 55 |
app.ml = ml
|
| 56 |
+
logger.info("Done with startup steps!")
|
| 57 |
return app
|
| 58 |
|
| 59 |
|