Update app.py

app.py

@@ -27,7 +27,6 @@ def generate_caption_cnnrnn(image):
     with open('Image_features_ecoder_decoder.pickle', 'rb') as f:
         Xnet_features = pickle.load(f)
         image = Xnet_features[image]
-        print(image.shape)
         caption = cnnrnn.get_result(image)
     return caption
 
@@ -75,6 +74,7 @@ def predict(img, model_name, max_tokens, temperature, imgID):
         return generate_caption_vitCoAtt(img),  getCaption(imgID)
     elif model_name == "Baseline Model CNN-RNN":
         img = getImageID(imgID)
+        print(img)
         return generate_caption_cnnrnn(img), getCaption(imgID)
     else:
         return "select a model","select an image"