basic

Author: Wauplin

packages/inference/src/snippets/templates/python/huggingface_hub/basic.jinja

@@ -1,5 +1,5 @@
 result = client.{{ methodName }}(
-    inputs={{ inputs }},
+    inputs={{ inputs.asObj.inputs }},
     model="{{ model.id }}",
 )
 

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

@@ -3,5 +3,5 @@ def query(payload):
     return response.json()
 
 output = query({
-    "inputs": {{ inputs }},
+    "inputs": {{ inputs.asObj.inputs }},
 }) 

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

@@ -4,4 +4,4 @@ def query(filename):
     response = requests.post(API_URL, headers=headers, data=data)
     return response.json()
 
-output = query({{ inputs }}) 
+output = query({{ inputs.asObj.inputs }})

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

@@ -156,17 +156,17 @@ const TEST_CASES: {
 		providers: ["replicate", "fal-ai"],
 		languages: ["js", "py"],
 	},
-	// {
-	// 	testName: "text-classification",
-	// 	model: {
-	// 		id: "distilbert/distilbert-base-uncased-finetuned-sst-2-english",
-	// 		pipeline_tag: "text-classification",
-	// 		tags: [],
-	// 		inference: "",
-	// 	},
-	// 	providers: ["hf-inference"],
-	// 	languages: ["sh", "js", "py"],
-	// },
+	{
+		testName: "text-classification",
+		model: {
+			id: "distilbert/distilbert-base-uncased-finetuned-sst-2-english",
+			pipeline_tag: "text-classification",
+			tags: [],
+			inference: "",
+		},
+		providers: ["hf-inference"],
+		languages: ["sh", "js", "py"],
+	},
 	// {
 	// 	testName: "basic-snippet--token-classification",
 	// 	model: {