fashxp/zero-shot-image-classification

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"
  }
}