ydshieh
/

flax-vision-encoder-decoder-vit-gpt2-coco-en

Model card Files Files and versions

ydshieh commited on Dec 20, 2021

Commit

0543ba7

·

1 Parent(s): d3f8014

fix

Files changed (1) hide show

run_image_captioning_flax.py +2 -1

run_image_captioning_flax.py CHANGED Viewed

@@ -738,7 +738,8 @@ def main():
         bools = []
         for image_file in examples[image_column]:
             try:
-                Image.open(image_file)
                 bools.append(True)
             except:
                 bools.append(False)

         bools = []
         for image_file in examples[image_column]:
             try:
+                image = Image.open(image_file)
+                encoder_inputs = feature_extractor(images=image, return_tensors="np")
                 bools.append(True)
             except:
                 bools.append(False)