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('.'),
|