Hugging Face's logo

Spaces:
1bincooo
/
rerank
Sleeping

App Files Community
1bincooo commited on
Commit
195b7ea
·
verified ·
1 Parent(s): e6c7cf3

Update localrerank.py

Browse files
Files changed (1) hide show
  1. localrerank.py +1 -1
localrerank.py CHANGED
@@ -12,7 +12,7 @@ app = FastAPI()
12
  security = HTTPBearer()
13
 
14
  SK_KEY = os.environ.get("SK_KEY", "sk-aaabbbcccdddeeefffggghhhiiijjjkkk")
15
- MODEL_ID = os.environ.get("RERANK_MODEL", "Qwen/Qwen3-Reranker-8B")
16
  MAX_LENGTH = int(os.environ.get("MAX_LENGTH", "1024"))
17
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
18
 
 
12
  security = HTTPBearer()
13
 
14
  SK_KEY = os.environ.get("SK_KEY", "sk-aaabbbcccdddeeefffggghhhiiijjjkkk")
15
+ MODEL_ID = os.environ.get("RERANK_MODEL", "Qwen/Qwen3-Reranker-4B")
16
  MAX_LENGTH = int(os.environ.get("MAX_LENGTH", "1024"))
17
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
18