Blaxzter
/

LaBSE-sentence-embeddings

Sentence Similarity

feature-extraction

sentence_embedding

Inference Endpoints

text-embeddings-inference

Model card Files Files and versions Community

Blaxzter commited on May 4, 2023

Commit

d42ba8d

•

1 Parent(s): 06ffed8

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -22,7 +22,7 @@ class EndpointHandler():
         for index, input_string in enumerate(inputs):
             if input_string in self.cache:
-                cached_results[index] = self.cache[input_string]
             else:
                 uncached_inputs.append((index, input_string))

         for index, input_string in enumerate(inputs):
             if input_string in self.cache:
+                cached_results[index] = self.cache[input_string]["pooler_output"]
             else:
                 uncached_inputs.append((index, input_string))