Spaces:
Running
Running
Commit
·
91fb3b7
1
Parent(s):
13cbb93
merged
Browse files
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'."}
|