SebastianSchramm commited on
Commit
aaec9ae
·
1 Parent(s): 0623b01

add startup logs

Browse files
Files changed (1) hide show
  1. server.py +4 -0
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