Hugging Face's logo

keras-io
/
image-captioning

Image-to-Text
TF-Keras
generic
Model card Files Community
merve HF staff commited on
Commit
3976308
•
1 Parent(s): b6683cb

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -2
pipeline.py CHANGED
@@ -24,8 +24,7 @@ class PreTrainedPipeline():
24
 
25
 
26
 
27
- def load_image(image_path):
28
- img = tf.io.read_file(image_path)
29
  img = tf.io.decode_jpeg(img, channels=3)
30
  img = tf.image.resize(img, (299, 299))
31
  img = tf.keras.applications.inception_v3.preprocess_input(img)
 
24
 
25
 
26
 
27
+ def load_image(img):
 
28
  img = tf.io.decode_jpeg(img, channels=3)
29
  img = tf.image.resize(img, (299, 299))
30
  img = tf.keras.applications.inception_v3.preprocess_input(img)