evgensoft commited on
Commit
55d019d
1 Parent(s): 1849912

Update m3_server.py

Browse files
Files changed (1) hide show
  1. m3_server.py +5 -5
m3_server.py CHANGED
@@ -185,7 +185,7 @@ description = dedent(
185
  embed example:
186
  ```bash
187
  curl -X 'POST' \
188
- 'https://mikeee-baai-m3.hf.space/embed/' \
189
  -H 'accept: application/json' \
190
  -H 'Content-Type: application/json' \
191
  -d '{
@@ -202,14 +202,14 @@ description = dedent(
202
  )
203
 
204
  app = FastAPI(
205
- title="Serving BAAI/bge-m3 embed and rerank",
206
- # description="Swagger UI at https://mikeee-baai-m3.hf.space/docs",
207
  description=description,
208
  version="0.1.0a0",
209
  )
210
 
211
  # Initialize the model and request processor
212
- model = m3Wrapper("BAAI/bge-m3")
213
  processor = RequestProcessor(
214
  model, accumulation_timeout=request_flush_timeout, max_request_to_flush=max_request
215
  )
@@ -235,7 +235,7 @@ async def timeout_middleware(request: Request, call_next):
235
  @app.get("/")
236
  async def landing():
237
  """Define landing page."""
238
- # return "Swagger UI at https://mikeee-baai-m3.hf.space/docs"
239
  return RedirectResponse("/docs")
240
 
241
 
 
185
  embed example:
186
  ```bash
187
  curl -X 'POST' \
188
+ 'https://evgensoft-baai-m3.hf.space/embed/' \
189
  -H 'accept: application/json' \
190
  -H 'Content-Type: application/json' \
191
  -d '{
 
202
  )
203
 
204
  app = FastAPI(
205
+ title="Serving deepvk/USER-bge-m3 embed and rerank",
206
+ # description="Swagger UI at https://evgensoft-baai-m3.hf.space/docs",
207
  description=description,
208
  version="0.1.0a0",
209
  )
210
 
211
  # Initialize the model and request processor
212
+ model = m3Wrapper("deepvk/USER-bge-m3")
213
  processor = RequestProcessor(
214
  model, accumulation_timeout=request_flush_timeout, max_request_to_flush=max_request
215
  )
 
235
  @app.get("/")
236
  async def landing():
237
  """Define landing page."""
238
+ # return "Swagger UI at https://evgensoft-baai-m3.hf.space/docs"
239
  return RedirectResponse("/docs")
240
 
241