zero shot class

Author: Wauplin

packages/inference/src/snippets/templates/python/requests/zeroShotClassification.jinja

@@ -3,6 +3,6 @@ def query(payload):
     return response.json()
 
 output = query({
-    "inputs": {{ inputs }},
+    "inputs": {{ inputs.asObj.inputs }},
     "parameters": {"candidate_labels": ["refund", "legal", "faq"]},
 }) 

packages/inference/src/snippets/templates/python/requests/zeroShotImageClassification.jinja

@@ -9,6 +9,6 @@ def query(data):
     return response.json()
 
 output = query({
-    "image_path": {{ inputs }},
+    "image_path": {{ inputs.asObj.inputs }},
     "parameters": {"candidate_labels": ["cat", "dog", "llama"]},
 }) 

packages/tasks-gen/scripts/generate-snippets-fixtures.ts

@@ -178,17 +178,17 @@ const TEST_CASES: {
 		providers: ["hf-inference"],
 		languages: ["py"],
 	},
-	// {
-	// 	testName: "zero-shot-classification",
-	// 	model: {
-	// 		id: "facebook/bart-large-mnli",
-	// 		pipeline_tag: "zero-shot-classification",
-	// 		tags: [],
-	// 		inference: "",
-	// 	},
-	// 	providers: ["hf-inference"],
-	// 	languages: ["py"],
-	// },
+	{
+		testName: "zero-shot-classification",
+		model: {
+			id: "facebook/bart-large-mnli",
+			pipeline_tag: "zero-shot-classification",
+			tags: [],
+			inference: "",
+		},
+		providers: ["hf-inference"],
+		languages: ["py"],
+	},
 	// {
 	// 	testName: "zero-shot-image-classification",
 	// 	model: {

packages/tasks-gen/snippets-fixtures/zero-shot-classification/0.requests.hf-inference.py

@@ -0,0 +1,13 @@
+import requests
+
+API_URL = "https://router.huggingface.co/hf-inference/models/facebook/bart-large-mnli"
+headers = {"Authorization": "Bearer api_token"}
+
+def query(payload):
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.json()
+
+output = query({
+    "inputs": "Hi, I recently bought a device from your company but it is not working as advertised and I would like to get reimbursed!",
+    "parameters": {"candidate_labels": ["refund", "legal", "faq"]},
+})