Anni01 commited on
Commit
a529a7e
·
verified ·
1 Parent(s): e2cfd01

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +3 -3
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
- import nltk
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,