fashxp's picture
Update README.md
bab69ce verified
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"
  }
}