Add Zero-shot classification to Google Sheets using HuggingFace's API

gistfile1.txt

function ANALYZE(input, labels, repo_id="facebook/bart-large-mnli") {

  endpoint = "https://api-inference.huggingface.co/models/" + repo_id;

  const payload = JSON.stringify({
    "inputs": input,
    "parameters": {"candidate_labels": labels}});

  // Add your token from https://huggingface.co/settings/token
  const options = {
      "headers":  {"Authorization": "Bearer <YOUR HUGGINGFACE API KEY>"},
      "wait_for_model": true,
      "use_gpu": false,
      "method" : "POST",
      "contentType" : "application/json",
      "payload" : payload,
      
  };

  const response = UrlFetchApp.fetch(endpoint, options);

  const data = JSON.parse(response.getContentText());

  return data['labels'][0];
}