Hugging Face's logo

optimum-intel
/
fastrag-ranker

Inference Endpoints
Model card Files Community
IlyasMoutawwakil HF staff commited on
Commit
4f8ccb0
·
verified ·
1 Parent(s): 512293f

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -3,7 +3,7 @@ from typing import Any, Dict, List
3
  from fastrag.rankers import QuantizedBiEncoderRanker
4
 
5
 
6
- class RankerHandler:
7
  def __init__(self, path=""):
8
  model_id = "Intel/bge-large-en-v1.5-rag-int8-static"
9
  self.ranker = QuantizedBiEncoderRanker(model_name_or_path=model_id)
 
3
  from fastrag.rankers import QuantizedBiEncoderRanker
4
 
5
 
6
+ class EndpointHandler:
7
  def __init__(self, path=""):
8
  model_id = "Intel/bge-large-en-v1.5-rag-int8-static"
9
  self.ranker = QuantizedBiEncoderRanker(model_name_or_path=model_id)