bipin
/

image-caption-generator

vision-encoder-decoder

image-captioning

Inference Endpoints

Model card Files Files and versions Community

bipin commited on Jul 27

Commit

f92c95b

•

1 Parent(s): 399b8b2

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -41,9 +41,9 @@ It achieves the following results on the evaluation set:
     model.to(device)
     ```
-2. Load the image for which the caption is to be generated
     ```python
-    ### replace the file name with image of your choice
     img_name = "flickr_data.jpg"
     img = Image.open(img_name)
     if img.mode != 'RGB':

     model.to(device)
     ```
+2. Load the image for which the caption is to be generated(note: replace the value of `img_name` with image of your choice)
     ```python
+    ### replace the value with your image
     img_name = "flickr_data.jpg"
     img = Image.open(img_name)
     if img.mode != 'RGB':