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 |
|