Update cnnrnn.py

cnnrnn.py

@@ -32,10 +32,10 @@ from nltk.translate.bleu_score import sentence_bleu
 
 
 
-def getModel():
+def getModel(image):
     embedding_matrix_vocab = np.load('my_embedding_matrix.npy')
 
-    input1 = Input(shape=(2048), name='Image_input')
+    input1 = Input(shape=(image.shape[1]), name='Image_input')
     dense1 = Dense(256, kernel_initializer=tf.keras.initializers.glorot_uniform(seed = 56), name='dense_encoder')(input1)
 
     input2 = Input(shape=(153), name='Text_Input')
@@ -93,7 +93,7 @@ def greedysearch(image):
     # Open the pickle file for reading
     train_data = pd.read_csv('Final_Train_Data.csv')
     y_train = train_data['Report']
-    encoder_model, decoder_model = getModel()
+    encoder_model, decoder_model = getModel(image)
     input_ = 'startseq'
     image_features = encoder_model.predict(image)
     result = []