fiewolf1000 commited on
Commit
abceca3
·
verified ·
1 Parent(s): 6e6c6a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -76,7 +76,7 @@ class RerankRequest(BaseModel):
76
  class DocumentScore(BaseModel):
77
  document: str
78
  relevance_score: float
79
- rank: int
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), rank=i+1)
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: