Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -129,8 +129,8 @@ def display_UI():
|
|
| 129 |
new_model = ImageCaptioningModel(cnn_model=cnn_model, encoder=encoder, decoder=decoder, image_aug=image_augmentation)
|
| 130 |
# Call the model with some dummy input data to create its variables
|
| 131 |
dummy_img = tf.zeros((1, 229, 229, 3))
|
| 132 |
-
dummy_seq = tf.zeros((1, 5,
|
| 133 |
-
new_model(
|
| 134 |
|
| 135 |
# Load the weights
|
| 136 |
new_model.load_weights('model_weights.h5')
|
|
|
|
| 129 |
new_model = ImageCaptioningModel(cnn_model=cnn_model, encoder=encoder, decoder=decoder, image_aug=image_augmentation)
|
| 130 |
# Call the model with some dummy input data to create its variables
|
| 131 |
dummy_img = tf.zeros((1, 229, 229, 3))
|
| 132 |
+
dummy_seq = tf.zeros((1, 5, 512))
|
| 133 |
+
new_model((dummy_img, dummy_seq))
|
| 134 |
|
| 135 |
# Load the weights
|
| 136 |
new_model.load_weights('model_weights.h5')
|