keras-io
/

image-captioning

Model card Files Files and versions

merve HF Staff commited on Dec 7, 2021

Commit

7ed71e4

·

1 Parent(s): 35e561e

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -10,7 +10,7 @@ class PreTrainedPipeline():
         # load the model
         self.decoder = keras.models.load_model(os.path.join(path, "decoder"))
-        self.decoder = keras.models.load_model(os.path.join(path, "encoder"))
         image_model = tf.keras.applications.InceptionV3(include_top=False,
                                                 weights='imagenet')

         # load the model
         self.decoder = keras.models.load_model(os.path.join(path, "decoder"))
+        self.encoder = keras.models.load_model(os.path.join(path, "encoder"))
         image_model = tf.keras.applications.InceptionV3(include_top=False,
                                                 weights='imagenet')