text-classification

Author: Wauplin

packages/tasks-gen/snippets-fixtures/text-classification/0.curl.hf-inference.sh

@@ -0,0 +1,5 @@
+curl https://router.huggingface.co/hf-inference/models/distilbert/distilbert-base-uncased-finetuned-sst-2-english \
+	-X POST \
+	-d '{"inputs": "I like you. I love you"}' \
+	-H 'Content-Type: application/json' \
+	-H 'Authorization: Bearer api_token'

packages/tasks-gen/snippets-fixtures/text-classification/0.huggingface.js.hf-inference.js

@@ -0,0 +1,11 @@
+import { InferenceClient } from "@huggingface/inference";
+
+const client = new InferenceClient("api_token");
+
+const output = await client.textClassification({
+	model: "distilbert/distilbert-base-uncased-finetuned-sst-2-english",
+	inputs: "I like you. I love you",
+	provider: "hf-inference",
+});
+
+console.log(output);

packages/tasks-gen/snippets-fixtures/text-classification/0.huggingface_hub.hf-inference.py

@@ -0,0 +1,13 @@
+from huggingface_hub import InferenceClient
+
+client = InferenceClient(
+    provider="hf-inference",
+    api_key="api_token",
+)
+
+result = client.text_classification(
+    inputs="I like you. I love you",
+    model="distilbert/distilbert-base-uncased-finetuned-sst-2-english",
+)
+
+print(result)

packages/tasks-gen/snippets-fixtures/text-classification/1.fetch.hf-inference.js

@@ -0,0 +1,19 @@
+async function query(data) {
+	const response = await fetch(
+		"https://router.huggingface.co/hf-inference/models/distilbert/distilbert-base-uncased-finetuned-sst-2-english",
+		{
+			headers: {
+				Authorization: "Bearer api_token",
+				"Content-Type": "application/json",
+			},
+			method: "POST",
+			body: JSON.stringify(data),
+		}
+	);
+	const result = await response.json();
+	return result;
+}
+
+query({"inputs": "I like you. I love you"}).then((response) => {
+	console.log(JSON.stringify(response));
+});

packages/tasks-gen/snippets-fixtures/text-classification/1.requests.hf-inference.py

@@ -0,0 +1,12 @@
+import requests
+
+API_URL = "https://router.huggingface.co/hf-inference/models/distilbert/distilbert-base-uncased-finetuned-sst-2-english"
+headers = {"Authorization": "Bearer api_token"}
+
+def query(payload):
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.json()
+
+output = query({
+    "inputs": "I like you. I love you",
+})