Jyothirmai commited on
Commit
b0643c7
·
verified ·
1 Parent(s): 60e40d8

Update cnnrnn.py

Browse files
Files changed (1) hide show
  1. cnnrnn.py +2 -1
cnnrnn.py CHANGED
@@ -35,7 +35,8 @@ from nltk.translate.bleu_score import sentence_bleu
35
  def getModel(image):
36
  embedding_matrix_vocab = np.load('my_embedding_matrix.npy')
37
 
38
- input1 = Input(shape=(image.shape[1]), name='Image_input')
 
39
  dense1 = Dense(256, kernel_initializer=tf.keras.initializers.glorot_uniform(seed = 56), name='dense_encoder')(input1)
40
 
41
  input2 = Input(shape=(153), name='Text_Input')
 
35
  def getModel(image):
36
  embedding_matrix_vocab = np.load('my_embedding_matrix.npy')
37
 
38
+ input_shape = (image.shape[1],)
39
+ input1 = Input(shape=input_shape, name='Image_input')
40
  dense1 = Dense(256, kernel_initializer=tf.keras.initializers.glorot_uniform(seed = 56), name='dense_encoder')(input1)
41
 
42
  input2 = Input(shape=(153), name='Text_Input')