hoangthan
/

img-caption-demo

Model card Files Files and versions

hoangthan commited on Aug 20, 2023

Commit

1199618

·

1 Parent(s): a134d0c

Update pipeline.py

Files changed (1) hide show

pipeline.py +2 -1

pipeline.py CHANGED Viewed

@@ -36,7 +36,8 @@ class PreTrainedPipeline():
             A :obj:`list`:. The list contains items that are dicts should be liked {"label": "XXX", "score": 0.82}
                 It is preferred if the returned list is in decreasing `score` order
         """
-        image = load_image(inputs)
         initial = self.word_to_index([['[START]']]) # (batch, sequence)
         img_features = self.model.feature_extractor(image[tf.newaxis, ...])
         temperature = 0

             A :obj:`list`:. The list contains items that are dicts should be liked {"label": "XXX", "score": 0.82}
                 It is preferred if the returned list is in decreasing `score` order
         """
+        img_array = tf.keras.utils.img_to_array(inputs)
+        image = load_image(img_array)
         initial = self.word_to_index([['[START]']]) # (batch, sequence)
         img_features = self.model.feature_extractor(image[tf.newaxis, ...])
         temperature = 0