Narya-ai
/

lr-bert-base-uncased

Inference Endpoints

Model card Files Files and versions Community

Uan Sholanbayev commited on Oct 13, 2023

Commit

a22dbf1

·

1 Parent(s): 8f18779

fix output format

Files changed (1) hide show

handler.py +3 -1

handler.py CHANGED Viewed

@@ -33,4 +33,6 @@ class EndpointHandler():
         texts_vec = self.get_embeddings(texts)
         diff = (np.array(texts_vec)[:, np.newaxis] - np.array(queries_vec))\
             .reshape(-1, len(queries_vec[0]))
-        return self.model.predict_proba(diff)

         texts_vec = self.get_embeddings(texts)
         diff = (np.array(texts_vec)[:, np.newaxis] - np.array(queries_vec))\
             .reshape(-1, len(queries_vec[0]))
+        return [{
+            "outputs": self.model.predict_proba(diff)
+        }]