Update app.py
Browse files
app.py
CHANGED
|
@@ -75,7 +75,7 @@ def train_val_split(caption_data, train_size=0.8, shuffle=True):
|
|
| 75 |
img_name: caption_data[img_name] for img_name in all_images[train_size:]
|
| 76 |
}
|
| 77 |
return training_data, validation_data
|
| 78 |
-
captions_mapping, text_data = load_captions_data(RIPIOS.token.txt)
|
| 79 |
train_data, valid_data = train_val_split(captions_mapping)
|
| 80 |
|
| 81 |
|
|
|
|
| 75 |
img_name: caption_data[img_name] for img_name in all_images[train_size:]
|
| 76 |
}
|
| 77 |
return training_data, validation_data
|
| 78 |
+
captions_mapping, text_data = load_captions_data("RIPIOS.token.txt")
|
| 79 |
train_data, valid_data = train_val_split(captions_mapping)
|
| 80 |
|
| 81 |
|