ffreemt commited on
Commit
7f5a78c
1 Parent(s): 2999f6e

Update landing in m3_server.py

Browse files
Files changed (1) hide show
  1. m3_server.py +10 -1
m3_server.py CHANGED
@@ -124,7 +124,11 @@ class RequestProcessor:
124
  except Exception as e:
125
  raise HTTPException(status_code=500, detail=f"Internal Server Error {e}")
126
 
127
- app = FastAPI()
 
 
 
 
128
 
129
  # Initialize the model and request processor
130
  model = m3Wrapper('BAAI/bge-m3')
@@ -143,6 +147,11 @@ async def timeout_middleware(request: Request, call_next):
143
  'processing_time': process_time},
144
  status_code=HTTP_504_GATEWAY_TIMEOUT)
145
 
 
 
 
 
 
146
  @app.post("/embeddings/", response_model=EmbedResponse)
147
  async def get_embeddings(request: EmbedRequest):
148
  embeddings = await processor.process_request(request, 'embed')
 
124
  except Exception as e:
125
  raise HTTPException(status_code=500, detail=f"Internal Server Error {e}")
126
 
127
+ app = FastAPI(
128
+ title="baai m3, serving embed and rerank",
129
+ description="Swagger UI at https://mikeee-baai-m3.hf.space/docs",
130
+ version="0.1.0a0",
131
+ )
132
 
133
  # Initialize the model and request processor
134
  model = m3Wrapper('BAAI/bge-m3')
 
147
  'processing_time': process_time},
148
  status_code=HTTP_504_GATEWAY_TIMEOUT)
149
 
150
+ @app.get("/")
151
+ async def landing():
152
+ """Define landing page."""
153
+ return "Swagger UI at https://mikeee-baai-m3.hf.space/docs"
154
+
155
  @app.post("/embeddings/", response_model=EmbedResponse)
156
  async def get_embeddings(request: EmbedRequest):
157
  embeddings = await processor.process_request(request, 'embed')