Skip to main content
image to text

vit-gpt2-image-captioning

by nlpconnect

tags:

84,303 downloads941 likes
Get API Key

API Usage

Endpoint

POST https://api.gpu.fm/v1/inference

Request

curl -X POST https://api.gpu.fm/v1/inference \
  -H "Authorization: Bearer YOUR_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "nlpconnect/vit-gpt2-image-captioning",
    "inputs": "Your input here",
    "parameters": {
      "max_new_tokens": 256
    }
  }'

Response

{
  "result": "Model output",
  "usage": {
    "inputTokens": 12,
    "outputTokens": 128
  },
  "pricing": {
    "total": "0.000018"
  }
}

Python SDK

import requests

response = requests.post(
    "https://api.gpu.fm/v1/inference",
    headers={"Authorization": "Bearer YOUR_API_KEY"},
    json={
        "model": "nlpconnect/vit-gpt2-image-captioning",
        "inputs": "Your input here",
        "parameters": {"max_new_tokens": 256}
    }
)

result = response.json()
print(result["result"])

Model Documentation

Full documentation available on HuggingFace →

Pricing

Per Request$0.0007/request

Billed from your prepaid credits. No minimum commitment.

Details

Model ID
nlpconnect/vit-gpt2-image-captioning
Library
transformers
Pipeline
image to text
Updated
2/27/2023

Tags

transformerspytorchvision-encoder-decoderimage-text-to-textimage-to-textimage-captioningdoi:10.57967/hf/0222license:apache-2.0endpoints_compatibleregion:us