Update main.py

main.py

@@ -67,7 +67,6 @@ You will use the [MS-COCO dataset](http://cocodataset.org/#home) to train your m
 
 **Caution: large download ahead**. You'll use the training set, which is a 13GB file.
 """
-# os.system("wget http://images.cocodataset.org/zips/train2014.zip")
 # Download caption annotation files
 annotation_folder = '/annotations/'
 annotation_zip = tf.keras.utils.get_file('captions.zip',
@@ -78,13 +77,14 @@ annotation_file = os.path.dirname(annotation_zip)+'/annotations/captions_train20
 # os.remove(annotation_zip)
 
 # Download image files
+os.system("wget http://images.cocodataset.org/zips/train2014.zip")
 image_folder = '/train2014/'
 
-image_zip = tf.keras.utils.get_file('train2014.zip',
-                                      cache_subdir=os.path.abspath('.'),
-                                      origin='http://images.cocodataset.org/zips/train2014.zip',
-                                      extract=True)
-PATH = os.path.dirname(image_zip) + image_folder
+# image_zip = tf.keras.utils.get_file('train2014.zip',
+#                                       cache_subdir=os.path.abspath('.'),
+#                                       origin='http://images.cocodataset.org/zips/train2014.zip',
+#                                       extract=True)
+PATH = os.path.dirname("train2014.zip") + image_folder
 # os.remove(image_zip)