Update tasks/text.py

tasks/text.py

@@ -129,8 +129,7 @@ async def evaluate_text(request: TextEvaluationRequest):
             # score shape == (batch_size, max_length, 1)
             # we get 1 at the last axis because we are applying score to self.V
             # the shape of the tensor before applying self.V is (batch_size, max_length, units)
-            score = tf.nn.tanh(
-                self.W1(features) + self.W2(hidden_with_time_axis))
+            score = tf.nn.tanh(self.W1(features) + self.W2(hidden_with_time_axis))
             
             # attention_weights shape == (batch_size, max_length, 1)
             attention_weights = tf.nn.softmax(self.V(score), axis=1)
@@ -151,7 +150,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     
     (lstm, forward_h, forward_c, backward_h, backward_c) = Bidirectional(LSTM(RNN_CELL_SIZE, return_sequences=True, return_state=True), name="bi_lstm_1")(lstm)
 
-        state_h = Concatenate()([forward_h, backward_h])
+    state_h = Concatenate()([forward_h, backward_h])
     state_c = Concatenate()([forward_c, backward_c])
     
     context_vector, attention_weights = Attention(10)(lstm, state_h)