fashxp's picture
initial commit
2fb700d
|
raw
history blame
847 Bytes
metadata
tags:
  - vision
  - zero-shot-image-classification
  - endpoints-template
inference: true
pipeline_tag: zero-shot-image-classification
base_model: openai/clip-vit-large-patch14
library_name: generic

Fork of openai/clip-vit-base-patch32 for a zero-sho-image-classification Inference endpoint.

This repository implements a custom task for zero-shot-image-classification for 🤗 Inference Endpoints. The code for the customized pipeline is in the handler.py.

To use deploy this model an Inference Endpoint you have to select Custom as task to use the handler.py file.

expected Request payload

{
  "image": encoded_image,
  "parameters": {
    "candidate_labels": "green, yellow, blue, white, silver"
  }
}

encoded_image is a base64 encoded image.