Update model.py
Browse files
model.py
CHANGED
|
@@ -15,7 +15,8 @@ import keras
|
|
| 15 |
from keras import layers
|
| 16 |
from keras.applications import efficientnet
|
| 17 |
from keras.layers import TextVectorization
|
| 18 |
-
|
|
|
|
| 19 |
keras.utils.set_random_seed(111)
|
| 20 |
|
| 21 |
|
|
@@ -221,8 +222,7 @@ class TransformerDecoderBlock(layers.Layer):
|
|
| 221 |
return tf.tile(mask, mult)
|
| 222 |
|
| 223 |
|
| 224 |
-
|
| 225 |
-
from nltk.translate.bleu_score import corpus_bleu
|
| 226 |
class ImageCaptioningModel(keras.Model):
|
| 227 |
def __init__(
|
| 228 |
self,
|
|
|
|
| 15 |
from keras import layers
|
| 16 |
from keras.applications import efficientnet
|
| 17 |
from keras.layers import TextVectorization
|
| 18 |
+
import nltk
|
| 19 |
+
from nltk.translate.bleu_score import corpus_bleu
|
| 20 |
keras.utils.set_random_seed(111)
|
| 21 |
|
| 22 |
|
|
|
|
| 222 |
return tf.tile(mask, mult)
|
| 223 |
|
| 224 |
|
| 225 |
+
|
|
|
|
| 226 |
class ImageCaptioningModel(keras.Model):
|
| 227 |
def __init__(
|
| 228 |
self,
|