Update README.md
Browse files
README.md
CHANGED
|
@@ -57,8 +57,6 @@ def predict_step(image_paths):
|
|
| 57 |
return preds
|
| 58 |
|
| 59 |
|
| 60 |
-
predict_step(['butterfly.jpg'])
|
| 61 |
-
|
| 62 |
```
|
| 63 |
|
| 64 |
# Sample running code using transformers pipeline
|
|
@@ -69,7 +67,7 @@ from transformers import pipeline
|
|
| 69 |
|
| 70 |
image_to_text = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
|
| 71 |
|
| 72 |
-
image_to_text("https://
|
| 73 |
|
| 74 |
# [{'generated_text': 'a soccer game with a player jumping to catch the ball '}]
|
| 75 |
|
|
|
|
| 57 |
return preds
|
| 58 |
|
| 59 |
|
|
|
|
|
|
|
| 60 |
```
|
| 61 |
|
| 62 |
# Sample running code using transformers pipeline
|
|
|
|
| 67 |
|
| 68 |
image_to_text = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
|
| 69 |
|
| 70 |
+
image_to_text("https://sneha.github.io/assets/images/image-captioning-example.png")
|
| 71 |
|
| 72 |
# [{'generated_text': 'a soccer game with a player jumping to catch the ball '}]
|
| 73 |
|