Update README.md
Browse files
README.md
CHANGED
|
@@ -41,9 +41,9 @@ It achieves the following results on the evaluation set:
|
|
| 41 |
model.to(device)
|
| 42 |
```
|
| 43 |
|
| 44 |
-
2. Load the image for which the caption is to be generated
|
| 45 |
```python
|
| 46 |
-
### replace the
|
| 47 |
img_name = "flickr_data.jpg"
|
| 48 |
img = Image.open(img_name)
|
| 49 |
if img.mode != 'RGB':
|
|
|
|
| 41 |
model.to(device)
|
| 42 |
```
|
| 43 |
|
| 44 |
+
2. Load the image for which the caption is to be generated(note: replace the value of `img_name` with image of your choice)
|
| 45 |
```python
|
| 46 |
+
### replace the value with your image
|
| 47 |
img_name = "flickr_data.jpg"
|
| 48 |
img = Image.open(img_name)
|
| 49 |
if img.mode != 'RGB':
|