IlyasMoutawwakil HF staff commited on
Commit
129944b
·
verified ·
1 Parent(s): 7fc99b0

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -12,8 +12,8 @@ class EndpointHandler:
12
  def __init__(self, path=""):
13
  model_id = "Intel/bge-small-en-v1.5-rag-int8-static"
14
 
15
- self.query_embedder = IPEXSentenceTransformersTextEmbedder(model_id)
16
- self.document_embedder = IPEXSentenceTransformersDocumentEmbedder(model_id)
17
 
18
  self.query_embedder.warm_up()
19
  self.document_embedder.warm_up()
 
12
  def __init__(self, path=""):
13
  model_id = "Intel/bge-small-en-v1.5-rag-int8-static"
14
 
15
+ self.query_embedder = IPEXSentenceTransformersTextEmbedder(model=model_id)
16
+ self.document_embedder = IPEXSentenceTransformersDocumentEmbedder(model=model_id)
17
 
18
  self.query_embedder.warm_up()
19
  self.document_embedder.warm_up()