Instructions to use PuppetLover/scenegraph_image_captioning with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use PuppetLover/scenegraph_image_captioning with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("PuppetLover/scenegraph_image_captioning") model = AutoModelForSeq2SeqLM.from_pretrained("PuppetLover/scenegraph_image_captioning", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -14,7 +14,7 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 14 |
|
| 15 |
# scenegraph_image_captioning
|
| 16 |
|
| 17 |
-
This model is a fine-tuned version of [facebook/mbart-large-50](https://huggingface.co/facebook/mbart-large-50) on an
|
| 18 |
It achieves the following results on the evaluation set:
|
| 19 |
- Loss: 3.3994
|
| 20 |
- Bleu4: 10.148
|
|
|
|
| 14 |
|
| 15 |
# scenegraph_image_captioning
|
| 16 |
|
| 17 |
+
This model is a fine-tuned version of [facebook/mbart-large-50](https://huggingface.co/facebook/mbart-large-50) on an `ViVG-Scene-Graph` dataset.
|
| 18 |
It achieves the following results on the evaluation set:
|
| 19 |
- Loss: 3.3994
|
| 20 |
- Bleu4: 10.148
|