Update main.py

main.py

@@ -78,6 +78,8 @@ annotation_file = os.path.dirname(annotation_zip)+'/annotations/captions_train20
 
 # Download image files
 os.system("wget http://images.cocodataset.org/zips/train2014.zip")
+os.system("unzip train2014.zip")
+
 image_folder = '/train2014/'
 
 # image_zip = tf.keras.utils.get_file('train2014.zip',