reichenbach commited on
Commit
21eea14
·
1 Parent(s): b1ba8e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ encoder_outputs, state_h_enc, state_c_enc = model.layers[2].output # lstm_1
53
  encoder_states = [state_h_enc, state_c_enc]
54
  encoder_model = keras.Model(encoder_inputs, encoder_states)
55
 
56
- decoder_inputs = model.input[1] # input_2
57
  decoder_state_input_h = keras.Input(shape=(latent_dim,))
58
  decoder_state_input_c = keras.Input(shape=(latent_dim,))
59
  decoder_states_inputs = [decoder_state_input_h, decoder_state_input_c]
 
53
  encoder_states = [state_h_enc, state_c_enc]
54
  encoder_model = keras.Model(encoder_inputs, encoder_states)
55
 
56
+ decoder_inputs = tf.identity(model.input[1]) # input_2
57
  decoder_state_input_h = keras.Input(shape=(latent_dim,))
58
  decoder_state_input_c = keras.Input(shape=(latent_dim,))
59
  decoder_states_inputs = [decoder_state_input_h, decoder_state_input_c]