Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -76,7 +76,7 @@ class RerankRequest(BaseModel):
|
|
| 76 |
class DocumentScore(BaseModel):
|
| 77 |
document: str
|
| 78 |
relevance_score: float
|
| 79 |
-
|
| 80 |
|
| 81 |
class RerankResponse(BaseModel):
|
| 82 |
request_id: str
|
|
@@ -189,7 +189,7 @@ class CrossEncoderModel:
|
|
| 189 |
logger.info(f"请求 {request_id}:重排序完成,返回 {len(sorted_docs)} 个结果")
|
| 190 |
|
| 191 |
return [
|
| 192 |
-
DocumentScore(document=doc, relevance_score=round(score, 4),
|
| 193 |
for i, (doc, score) in enumerate(sorted_docs)
|
| 194 |
]
|
| 195 |
except Exception as e:
|
|
|
|
| 76 |
class DocumentScore(BaseModel):
|
| 77 |
document: str
|
| 78 |
relevance_score: float
|
| 79 |
+
index: int
|
| 80 |
|
| 81 |
class RerankResponse(BaseModel):
|
| 82 |
request_id: str
|
|
|
|
| 189 |
logger.info(f"请求 {request_id}:重排序完成,返回 {len(sorted_docs)} 个结果")
|
| 190 |
|
| 191 |
return [
|
| 192 |
+
DocumentScore(document=doc, relevance_score=round(score, 4), index=i+1)
|
| 193 |
for i, (doc, score) in enumerate(sorted_docs)
|
| 194 |
]
|
| 195 |
except Exception as e:
|