optimum-intel
/

fastrag-retriever

Inference Endpoints

Model card Files Files and versions Community

IlyasMoutawwakil HF staff commited on May 10

Commit

73e7897

•

1 Parent(s): 6ded72f

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -3,7 +3,7 @@ from typing import Any, Dict, List
 from fastrag.retrievers import QuantizedBiEncoderRetriever
-class RetrieverHandler:
     def __init__(self, path=""):
         model_id = "Intel/bge-small-en-v1.5-rag-int8-static"
         self.retriever = QuantizedBiEncoderRetriever(embedding_model=model_id)

 from fastrag.retrievers import QuantizedBiEncoderRetriever
+class EndpointHandler:
     def __init__(self, path=""):
         model_id = "Intel/bge-small-en-v1.5-rag-int8-static"
         self.retriever = QuantizedBiEncoderRetriever(embedding_model=model_id)