Spaces:
Running
Running
fahmiaziz98
commited on
Commit
·
87ef7db
1
Parent(s):
5e48e1f
clean-up
Browse files- app.py +2 -8
- core/model_manager.py +1 -1
app.py
CHANGED
|
@@ -52,17 +52,12 @@ High-performance API for document reranking using multiple state-of-the-art mode
|
|
| 52 |
- Input: Query + documents + optional instruction
|
| 53 |
- Output: Ranked documents with relevance scores
|
| 54 |
|
| 55 |
-
**Warning**: Not
|
| 56 |
""",
|
| 57 |
version="1.0.0",
|
| 58 |
lifespan=lifespan
|
| 59 |
)
|
| 60 |
|
| 61 |
-
|
| 62 |
-
# -------------------------
|
| 63 |
-
# API Endpoints
|
| 64 |
-
# -------------------------
|
| 65 |
-
|
| 66 |
@app.post("/rerank", response_model=RerankResponse, tags=["Reranking"])
|
| 67 |
async def rerank_documents(request: RerankRequest):
|
| 68 |
"""
|
|
@@ -105,8 +100,7 @@ async def rerank_documents(request: RerankRequest):
|
|
| 105 |
|
| 106 |
model = model_manager.get_model(request.model_id)
|
| 107 |
original_indices, documents = zip(*valid_docs)
|
| 108 |
-
|
| 109 |
-
logger.info(f"Document: {list(documents)}")
|
| 110 |
scores = model.rerank(
|
| 111 |
query=request.query.strip(),
|
| 112 |
documents=list(documents),
|
|
|
|
| 52 |
- Input: Query + documents + optional instruction
|
| 53 |
- Output: Ranked documents with relevance scores
|
| 54 |
|
| 55 |
+
⚠️ **Warning**: Not for production use!.
|
| 56 |
""",
|
| 57 |
version="1.0.0",
|
| 58 |
lifespan=lifespan
|
| 59 |
)
|
| 60 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 61 |
@app.post("/rerank", response_model=RerankResponse, tags=["Reranking"])
|
| 62 |
async def rerank_documents(request: RerankRequest):
|
| 63 |
"""
|
|
|
|
| 100 |
|
| 101 |
model = model_manager.get_model(request.model_id)
|
| 102 |
original_indices, documents = zip(*valid_docs)
|
| 103 |
+
|
|
|
|
| 104 |
scores = model.rerank(
|
| 105 |
query=request.query.strip(),
|
| 106 |
documents=list(documents),
|
core/model_manager.py
CHANGED
|
@@ -128,7 +128,7 @@ class ModelManager:
|
|
| 128 |
"id": model_id,
|
| 129 |
"name": config.get("model_name"),
|
| 130 |
"type": config.get("model_type"),
|
| 131 |
-
"language": config.get("
|
| 132 |
"description": config.get("description"),
|
| 133 |
"repository": config.get("repository"),
|
| 134 |
"loaded": model.loaded if model else False
|
|
|
|
| 128 |
"id": model_id,
|
| 129 |
"name": config.get("model_name"),
|
| 130 |
"type": config.get("model_type"),
|
| 131 |
+
"language": config.get("languages"),
|
| 132 |
"description": config.get("description"),
|
| 133 |
"repository": config.get("repository"),
|
| 134 |
"loaded": model.loaded if model else False
|