Summarization
Transformers
PyTorch
English
t5
text2text-generation
Trained with AutoTrain
text-generation-inference
Instructions to use YtBig/tag-caption-v2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use YtBig/tag-caption-v2 with Transformers:
# Use a pipeline as a high-level helper # Warning: Pipeline type "summarization" is no longer supported in transformers v5. # You must load the model directly (see below) or downgrade to v4.x with: # 'pip install "transformers<5.0.0' from transformers import pipeline pipe = pipeline("summarization", model="YtBig/tag-caption-v2")# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("YtBig/tag-caption-v2") model = AutoModelForSeq2SeqLM.from_pretrained("YtBig/tag-caption-v2") - Notebooks
- Google Colab
- Kaggle
Model Trained Using AutoTrain
- Problem type: Summarization
- Model ID: 2343573836
- CO2 Emissions (in grams): 5.3317
Validation Metrics
- Loss: 2.315
- Rouge1: 39.655
- Rouge2: 17.178
- RougeL: 38.522
- RougeLsum: 38.554
- Gen Len: 12.059
Usage
You can use cURL to access this model:
$ curl -X POST -H "Authorization: Bearer YOUR_HUGGINGFACE_API_KEY" -H "Content-Type: application/json" -d '{"inputs": "I love AutoTrain"}' https://api-inference.huggingface.co/Alfred-o/autotrain-tag-figcaption-2343573836
- Downloads last month
- 7