From 6707828af26d25a9c54bec0f0d094f04c0c8f8c5 Mon Sep 17 00:00:00 2001
From: joe <zjiuzhou@gmail.com>
Date: Thu, 31 Jan 2019 10:17:30 +0800
Subject: [PATCH] fix bug

---
 extract_feature.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/extract_feature.py b/extract_feature.py
index 820d061..ec7f295 100644
--- a/extract_feature.py
+++ b/extract_feature.py
@@ -86,7 +86,7 @@ class BertVector:
     def encode(self, sentence):
         self.sentence_len = len(sentence)
         self.input_queue.put(sentence)
-        prediction = self.output_queue.get()['encodes'][0]
+        prediction = self.output_queue.get()['encodes']
         return prediction
 
     def queue_predict_input_fn(self):