keras-io
/

image-captioning

Model card Files Files and versions Community

merve HF staff commited on Dec 6, 2021

Commit

08060ee

•

1 Parent(s): 37d00a0

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -19,7 +19,7 @@ class PreTrainedPipeline():
         self.image_features_extract_model = tf.keras.Model(new_input, hidden_layer)
-        with open('tokenizer.pickle', 'rb') as handle:
             self.tokenizer = pickle.load(handle)

         self.image_features_extract_model = tf.keras.Model(new_input, hidden_layer)
+        with open('./tokenizer.pickle', 'rb') as handle:
             self.tokenizer = pickle.load(handle)