sushil3125 commited on
Commit
91fb3b7
·
1 Parent(s): 13cbb93
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -92,3 +92,9 @@ async def get_embedding(input: TextInput):
92
  else:
93
  embeddings = bge_small_model.encode(input.text)
94
  return {"embeddings": embeddings.tolist()}
 
 
 
 
 
 
 
92
  else:
93
  embeddings = bge_small_model.encode(input.text)
94
  return {"embeddings": embeddings.tolist()}
95
+
96
+
97
+ @app.get("/status")
98
+ async def status():
99
+ logger.warning(f"Unsupported model_name: {input.model_name}")
100
+ return {"error": "Unsupported model_name. Use 'BM', 'BG', or 'SPLADE'."}