Hugging Face's logo

Spaces:
mrinjera
/
SWERank
Paused

App Files Community
mrinjera commited on
Commit
f1df6cd
·
verified ·
1 Parent(s): daeff1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from pydantic import BaseModel
4
  from typing import Optional, List, Tuple
5
 
6
  # load RankListwiseOSLLM
7
- reranker = RankListwiseOSLLM("Salesforce/SweRankLLM-small", device="cpu")
8
 
9
  class RerankRequest(BaseModel):
10
  query: str
 
4
  from typing import Optional, List, Tuple
5
 
6
  # load RankListwiseOSLLM
7
+ reranker = RankListwiseOSLLM("Salesforce/SweRankLLM-small")
8
 
9
  class RerankRequest(BaseModel):
10
  query: str