Spaces:
Runtime error
Runtime error
Update main.py
Browse files
main.py
CHANGED
@@ -68,7 +68,7 @@ You will use the [MS-COCO dataset](http://cocodataset.org/#home) to train your m
|
|
68 |
**Caution: large download ahead**. You'll use the training set, which is a 13GB file.
|
69 |
"""
|
70 |
# Download caption annotation files
|
71 |
-
annotation_folder = '
|
72 |
annotation_zip = tf.keras.utils.get_file('captions.zip',
|
73 |
cache_subdir=os.path.abspath('.'),
|
74 |
origin='http://images.cocodataset.org/annotations/annotations_trainval2014.zip',
|
@@ -80,7 +80,7 @@ annotation_file = os.path.dirname(annotation_zip)+'/annotations/captions_train20
|
|
80 |
os.system("wget http://images.cocodataset.org/zips/train2014.zip")
|
81 |
os.system("unzip train2014.zip")
|
82 |
|
83 |
-
image_folder = '
|
84 |
|
85 |
# image_zip = tf.keras.utils.get_file('train2014.zip',
|
86 |
# cache_subdir=os.path.abspath('.'),
|
|
|
68 |
**Caution: large download ahead**. You'll use the training set, which is a 13GB file.
|
69 |
"""
|
70 |
# Download caption annotation files
|
71 |
+
annotation_folder = 'annotations/'
|
72 |
annotation_zip = tf.keras.utils.get_file('captions.zip',
|
73 |
cache_subdir=os.path.abspath('.'),
|
74 |
origin='http://images.cocodataset.org/annotations/annotations_trainval2014.zip',
|
|
|
80 |
os.system("wget http://images.cocodataset.org/zips/train2014.zip")
|
81 |
os.system("unzip train2014.zip")
|
82 |
|
83 |
+
image_folder = 'train2014/'
|
84 |
|
85 |
# image_zip = tf.keras.utils.get_file('train2014.zip',
|
86 |
# cache_subdir=os.path.abspath('.'),
|