added model

handler.py

@@ -36,5 +36,5 @@ class EndpointHandler():
         diff = (np.array(texts_vec)[:, np.newaxis] - np.array(queries_vec))\
             .reshape(-1, len(queries_vec[0]))
         return [{
-            "outputs": self.model.predict_proba(diff)
+            "outputs": self.model.predict_proba(diff).tolist()
         }]

main_test.py

@@ -42,10 +42,10 @@ semi_relevant_query = "Did Cristiano Ronaldo did a great work in Saudi league?"
 
 
 # init handler
-my_handler = EndpointHandler(path="./bert_lr.pkl")
+my_handler = EndpointHandler(path=".")
 
 # prepare sample payload
-relevant_payload = {"queries": [query], "texts": [summary]}
+relevant_payload = {"queries": [query], "texts": [summary] * 50}
 irrelevant_payload = {"queries": semi_relevant_query, "texts": [summary]}
 
 # test the handler

model.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:68d45e234eb4a928074dfd868cead0219ab85354cc53d20e772753c6bb9169d3
+size 440449768