team-indain-image-caption
/

hindi-image-captioning

vision-encoder-decoder

Inference Endpoints

Model card Files Files and versions Community

krypticmouse commited on Nov 24, 2021

Commit

78f18cc

•

1 Parent(s): 576debb

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -11,7 +11,7 @@ Here is how to use this model to caption an image of the Flickr8k dataset:
 import torch
 import requests
 from PIL import Image
-from transformers import ViTFeatureExtractor, AutoTokenizer,
                          VisionEncoderDecoderModel
 if torch.cuda.is_available():

 import torch
 import requests
 from PIL import Image
+from transformers import ViTFeatureExtractor, AutoTokenizer, \
                          VisionEncoderDecoderModel
 if torch.cuda.is_available():