@huggingface/tasks 0.0.6 → 0.0.7

package/src/snippets/python.ts

@@ -0,0 +1,114 @@
+import type { ModelData } from "../model-data.js";
+import type { PipelineType } from "../pipelines.js";
+import { getModelInputSnippet } from "./inputs.js";
+
+export const snippetZeroShotClassification = (model: ModelData): string =>
+	`def query(payload):
+	response = requests.post(API_URL, headers=headers, json=payload)
+	return response.json()
+
+output = query({
+    "inputs": ${getModelInputSnippet(model)},
+    "parameters": {"candidate_labels": ["refund", "legal", "faq"]},
+})`;
+
+export const snippetBasic = (model: ModelData): string =>
+	`def query(payload):
+	response = requests.post(API_URL, headers=headers, json=payload)
+	return response.json()
+	
+output = query({
+	"inputs": ${getModelInputSnippet(model)},
+})`;
+
+export const snippetFile = (model: ModelData): string =>
+	`def query(filename):
+    with open(filename, "rb") as f:
+        data = f.read()
+    response = requests.post(API_URL, headers=headers, data=data)
+    return response.json()
+
+output = query(${getModelInputSnippet(model)})`;
+
+export const snippetTextToImage = (model: ModelData): string =>
+	`def query(payload):
+	response = requests.post(API_URL, headers=headers, json=payload)
+	return response.content
+image_bytes = query({
+	"inputs": ${getModelInputSnippet(model)},
+})
+# You can access the image with PIL.Image for example
+import io
+from PIL import Image
+image = Image.open(io.BytesIO(image_bytes))`;
+
+export const snippetTextToAudio = (model: ModelData): string => {
+	// Transformers TTS pipeline and api-inference-community (AIC) pipeline outputs are diverged
+	// with the latest update to inference-api (IA).
+	// Transformers IA returns a byte object (wav file), whereas AIC returns wav and sampling_rate.
+	if (model.library_name === "transformers") {
+		return `def query(payload):
+	response = requests.post(API_URL, headers=headers, json=payload)
+	return response.content
+
+audio_bytes = query({
+	"inputs": ${getModelInputSnippet(model)},
+})
+# You can access the audio with IPython.display for example
+from IPython.display import Audio
+Audio(audio_bytes)`;
+	} else {
+		return `def query(payload):
+	response = requests.post(API_URL, headers=headers, json=payload)
+	return response.json()
+	
+audio, sampling_rate = query({
+	"inputs": ${getModelInputSnippet(model)},
+})
+# You can access the audio with IPython.display for example
+from IPython.display import Audio
+Audio(audio, rate=sampling_rate)`;
+	}
+};
+export const pythonSnippets: Partial<Record<PipelineType, (model: ModelData) => string>> = {
+	// Same order as in js/src/lib/interfaces/Types.ts
+	"text-classification": snippetBasic,
+	"token-classification": snippetBasic,
+	"table-question-answering": snippetBasic,
+	"question-answering": snippetBasic,
+	"zero-shot-classification": snippetZeroShotClassification,
+	translation: snippetBasic,
+	summarization: snippetBasic,
+	conversational: snippetBasic,
+	"feature-extraction": snippetBasic,
+	"text-generation": snippetBasic,
+	"text2text-generation": snippetBasic,
+	"fill-mask": snippetBasic,
+	"sentence-similarity": snippetBasic,
+	"automatic-speech-recognition": snippetFile,
+	"text-to-image": snippetTextToImage,
+	"text-to-speech": snippetTextToAudio,
+	"text-to-audio": snippetTextToAudio,
+	"audio-to-audio": snippetFile,
+	"audio-classification": snippetFile,
+	"image-classification": snippetFile,
+	"image-to-text": snippetFile,
+	"object-detection": snippetFile,
+	"image-segmentation": snippetFile,
+};
+
+export function getPythonInferenceSnippet(model: ModelData, accessToken: string): string {
+	const body =
+		model.pipeline_tag && model.pipeline_tag in pythonSnippets ? pythonSnippets[model.pipeline_tag]?.(model) ?? "" : "";
+
+	return `import requests
+
+API_URL = "https://api-inference.huggingface.co/models/${model.id}"
+headers = {"Authorization": ${accessToken ? `"Bearer ${accessToken}"` : `f"Bearer {API_TOKEN}"`}}
+
+${body}`;
+}
+
+export function hasPythonInferenceSnippet(model: ModelData): boolean {
+	return !!model.pipeline_tag && model.pipeline_tag in pythonSnippets;
+}

package/src/{audio-classification → tasks/audio-classification}/about.md

@@ -66,7 +66,7 @@ Speaker Identification is classifying the audio of the person speaking. Speakers
 
 ## Solving audio classification for your own data
 
-We have some great news! You can do fine-tuning (transfer learning) to train a well-performing model without requiring as much data. Pretrained models such as Wav2Vec2 and HuBERT exist. [Facebook's Wav2Vec2 XLS-R model](https://ai.facebook.com/blog/wav2vec-20-learning-the-structure-of-speech-from-raw-audio/) is a large multilingual model trained on 128 languages and with 436K hours of speech.
+We have some great news! You can do fine-tuning (transfer learning) to train a well-performing model without requiring as much data. Pretrained models such as Wav2Vec2 and HuBERT exist. [Facebook's Wav2Vec2 XLS-R model](https://huggingface.co/docs/transformers/model_doc/xlsr_wav2vec2) is a large multilingual model trained on 128 languages and with 436K hours of speech. Similarly, you can also use [OpenAI's Whisper](https://huggingface.co/docs/transformers/model_doc/whisper) trained on up to 4 Million hours of multilingual speech data for this task too!
 
 ## Useful Resources
 
@@ -82,4 +82,5 @@ Would you like to learn more about the topic? Awesome! Here you can find some cu
 
 ### Documentation
 
+- [Hugging Face Audio Course](https://huggingface.co/learn/audio-course/chapter4/introduction)
 - [Audio classification task guide](https://huggingface.co/docs/transformers/tasks/audio_classification)

package/src/{audio-classification → tasks/audio-classification}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [
@@ -64,8 +64,8 @@ const taskData: TaskDataCustom = {
 	],
 	spaces: [
 		{
-			description: "An application that can predict the language spoken in a given audio.",
-			id: "akhaliq/Speechbrain-audio-classification",
+			description: "An application that can classify music into different genre.",
+			id: "kurianbenoy/audioclassification",
 		},
 	],
 	summary:

package/src/{audio-to-audio → tasks/audio-to-audio}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{automatic-speech-recognition → tasks/automatic-speech-recognition}/about.md

@@ -25,7 +25,7 @@ import json
 import requests
 
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
-API_URL = "https://api-inference.huggingface.co/models/openai/whisper-large-v2"
+API_URL = "https://api-inference.huggingface.co/models/openai/whisper-large-v3"
 
 def query(filename):
     with open(filename, "rb") as f:
@@ -63,7 +63,7 @@ await inference.automaticSpeechRecognition({
 
 ## Solving ASR for your own data
 
-We have some great news! You can fine-tune (transfer learning) a foundational speech model on a specific language without tonnes of data. Pretrained models such as Whisper, Wav2Vec2-MMS and HuBERT exist. [OpenAI's Whisper model](https://huggingface.co/openai/whisper-large-v2) is a large multilingual model trained on 100+ languages and with 680K hours of speech.
+We have some great news! You can fine-tune (transfer learning) a foundational speech model on a specific language without tonnes of data. Pretrained models such as Whisper, Wav2Vec2-MMS and HuBERT exist. [OpenAI's Whisper model](https://huggingface.co/openai/whisper-large-v3) is a large multilingual model trained on 100+ languages and with 4 Million hours of speech.
 
 The following detailed [blog post](https://huggingface.co/blog/fine-tune-whisper) shows how to fine-tune a pre-trained Whisper checkpoint on labeled data for ASR. With the right data and strategy you can fine-tune a high-performant model on a free Google Colab instance too. We suggest to read the blog post for more info!
 
@@ -75,6 +75,7 @@ These events help democratize ASR for all languages, including low-resource lang
 
 ## Useful Resources
 
+- [Hugging Face Audio Course](https://huggingface.co/learn/audio-course/chapter5/introduction)
 - [Fine-tuning MetaAI's MMS Adapter Models for Multi-Lingual ASR](https://huggingface.co/blog/mms_adapters)
 - [Making automatic speech recognition work on large files with Wav2Vec2 in 🤗 Transformers](https://huggingface.co/blog/asr-chunking)
 - [Boosting Wav2Vec2 with n-grams in 🤗 Transformers](https://huggingface.co/blog/wav2vec2-with-ngram)

package/src/{automatic-speech-recognition → tasks/automatic-speech-recognition}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [
@@ -44,7 +44,7 @@ const taskData: TaskDataCustom = {
 	models: [
 		{
 			description: "A powerful ASR model by OpenAI.",
-			id: "openai/whisper-large-v2",
+			id: "openai/whisper-large-v3",
 		},
 		{
 			description: "A good generic ASR model by MetaAI.",
@@ -58,20 +58,20 @@ const taskData: TaskDataCustom = {
 	spaces: [
 		{
 			description: "A powerful general-purpose speech recognition application.",
-			id: "openai/whisper",
+			id: "hf-audio/whisper-large-v3",
 		},
 		{
 			description: "Fastest speech recognition application.",
 			id: "sanchit-gandhi/whisper-jax",
 		},
 		{
-			description: "An application that transcribes speeches in YouTube videos.",
-			id: "jeffistyping/Youtube-Whisperer",
+			description: "A high quality speech and text translation model by Meta.",
+			id: "facebook/seamless_m4t",
 		},
 	],
 	summary:
 		"Automatic Speech Recognition (ASR), also known as Speech to Text (STT), is the task of transcribing a given audio to text. It has many applications, such as voice user interfaces.",
-	widgetModels: ["openai/whisper-large-v2"],
+	widgetModels: ["openai/whisper-large-v3"],
 	youtubeId: "TksaY_FDgnk",
 };
 

package/src/{conversational → tasks/conversational}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{depth-estimation → tasks/depth-estimation}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{document-question-answering → tasks/document-question-answering}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{feature-extraction → tasks/feature-extraction}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [
@@ -29,12 +29,7 @@ const taskData: TaskDataCustom = {
 			},
 		],
 	},
-	metrics: [
-		{
-			description: "",
-			id: "",
-		},
-	],
+	metrics: [],
 	models: [
 		{
 			description: "A powerful feature extraction model for natural language processing tasks.",

package/src/{fill-mask → tasks/fill-mask}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{image-classification → tasks/image-classification}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{image-segmentation → tasks/image-segmentation}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{image-to-image → tasks/image-to-image}/about.md

@@ -2,22 +2,22 @@
 
 ### Style transfer
 
-One of the most popular use cases of image to image is the style transfer. Style transfer models can convert a regular photography into a painting in the style of a famous painter.
+One of the most popular use cases of image-to-image is style transfer. Style transfer models can convert a normal photography into a painting in the style of a famous painter.
 
 ## Task Variants
 
 ### Image inpainting
 
-Image inpainting is widely used during photography editing to remove unwanted objects, such as poles, wires or sensor
+Image inpainting is widely used during photography editing to remove unwanted objects, such as poles, wires, or sensor
 dust.
 
 ### Image colorization
 
-Old, black and white images can be brought up to life using an image colorization model.
+Old or black and white images can be brought up to life using an image colorization model.
 
 ### Super Resolution
 
-Super resolution models increase the resolution of an image, allowing for higher quality viewing and printing.
+Super-resolution models increase the resolution of an image, allowing for higher-quality viewing and printing.
 
 ## Inference
 
@@ -55,20 +55,21 @@ await inference.imageToImage({
 
 ## ControlNet
 
-Controlling outputs of diffusion models only with a text prompt is a challenging problem. ControlNet is a neural network type that provides an image based control to diffusion models. These controls can be edges or landmarks in an image.
+Controlling the outputs of diffusion models only with a text prompt is a challenging problem. ControlNet is a neural network model that provides image-based control to diffusion models. Control images can be edges or other landmarks extracted from a source image.
 
 Many ControlNet models were trained in our community event, JAX Diffusers sprint. You can see the full list of the ControlNet models available [here](https://huggingface.co/spaces/jax-diffusers-event/leaderboard).
 
 ## Most Used Model for the Task
 
-Pix2Pix is a popular model used for image to image translation tasks. It is based on a conditional-GAN (generative adversarial network) where instead of a noise vector a 2D image is given as input. More information about Pix2Pix can be retrieved from this [link](https://phillipi.github.io/pix2pix/) where the associated paper and the GitHub repository can be found.
+Pix2Pix is a popular model used for image-to-image translation tasks. It is based on a conditional-GAN (generative adversarial network) where instead of a noise vector a 2D image is given as input. More information about Pix2Pix can be retrieved from this [link](https://phillipi.github.io/pix2pix/) where the associated paper and the GitHub repository can be found.
 
-Below images show some of the examples shared in the paper that can be obtained using Pix2Pix. There are various cases this model can be applied on. It is capable of relatively simpler things, e.g. converting a grayscale image to its colored version. But more importantly, it can generate realistic pictures from rough sketches (can be seen in the purse example) or from painting-like images (can be seen in the street and facade examples below).
+The images below show some examples extracted from the Pix2Pix paper. This model can be applied to various use cases. It is capable of relatively simpler things, e.g., converting a grayscale image to its colored version. But more importantly, it can generate realistic pictures from rough sketches (can be seen in the purse example) or from painting-like images (can be seen in the street and facade examples below).
 
 ![Examples](https://huggingface.co/datasets/huggingfacejs/tasks/resolve/main/image-to-image/pix2pix_examples.jpg)
 
 ## Useful Resources
 
+- [Image-to-image guide with diffusers](https://huggingface.co/docs/diffusers/using-diffusers/img2img)
 - [Train your ControlNet with diffusers 🧨](https://huggingface.co/blog/train-your-controlnet)
 - [Ultra fast ControlNet with 🧨 Diffusers](https://huggingface.co/blog/controlnet)
 

package/src/{image-to-image → tasks/image-to-image}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{image-to-text → tasks/image-to-text}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{tasksData.ts → tasks/index.ts}

@@ -1,5 +1,4 @@
-import { type PipelineType, PIPELINE_DATA } from "./pipelines";
-import type { TaskDataCustom, TaskData } from "./Types";
+import { type PipelineType, PIPELINE_DATA } from "../pipelines";
 
 import audioClassification from "./audio-classification/data";
 import audioToAudio from "./audio-to-audio/data";
@@ -34,7 +33,82 @@ import videoClassification from "./video-classification/data";
 import visualQuestionAnswering from "./visual-question-answering/data";
 import zeroShotClassification from "./zero-shot-classification/data";
 import zeroShotImageClassification from "./zero-shot-image-classification/data";
-import { TASKS_MODEL_LIBRARIES } from "./const";
+
+import type { ModelLibraryKey } from "../model-libraries";
+
+/**
+ * Model libraries compatible with each ML task
+ */
+export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
+	"audio-classification": ["speechbrain", "transformers"],
+	"audio-to-audio": ["asteroid", "speechbrain"],
+	"automatic-speech-recognition": ["espnet", "nemo", "speechbrain", "transformers", "transformers.js"],
+	conversational: ["transformers"],
+	"depth-estimation": ["transformers"],
+	"document-question-answering": ["transformers"],
+	"feature-extraction": ["sentence-transformers", "transformers", "transformers.js"],
+	"fill-mask": ["transformers", "transformers.js"],
+	"graph-ml": ["transformers"],
+	"image-classification": ["keras", "timm", "transformers", "transformers.js"],
+	"image-segmentation": ["transformers", "transformers.js"],
+	"image-to-image": ["diffusers"],
+	"image-to-text": ["transformers.js"],
+	"image-to-video": ["diffusers"],
+	"video-classification": ["transformers"],
+	"mask-generation": ["transformers"],
+	"multiple-choice": ["transformers"],
+	"object-detection": ["transformers", "transformers.js"],
+	other: [],
+	"question-answering": ["adapter-transformers", "allennlp", "transformers", "transformers.js"],
+	robotics: [],
+	"reinforcement-learning": ["transformers", "stable-baselines3", "ml-agents", "sample-factory"],
+	"sentence-similarity": ["sentence-transformers", "spacy", "transformers.js"],
+	summarization: ["transformers", "transformers.js"],
+	"table-question-answering": ["transformers"],
+	"table-to-text": ["transformers"],
+	"tabular-classification": ["sklearn"],
+	"tabular-regression": ["sklearn"],
+	"tabular-to-text": ["transformers"],
+	"text-classification": ["adapter-transformers", "spacy", "transformers", "transformers.js"],
+	"text-generation": ["transformers", "transformers.js"],
+	"text-retrieval": [],
+	"text-to-image": ["diffusers"],
+	"text-to-speech": ["espnet", "tensorflowtts", "transformers"],
+	"text-to-audio": ["transformers"],
+	"text-to-video": ["diffusers"],
+	"text2text-generation": ["transformers", "transformers.js"],
+	"time-series-forecasting": [],
+	"token-classification": [
+		"adapter-transformers",
+		"flair",
+		"spacy",
+		"span-marker",
+		"stanza",
+		"transformers",
+		"transformers.js",
+	],
+	translation: ["transformers", "transformers.js"],
+	"unconditional-image-generation": ["diffusers"],
+	"visual-question-answering": ["transformers"],
+	"voice-activity-detection": [],
+	"zero-shot-classification": ["transformers", "transformers.js"],
+	"zero-shot-image-classification": ["transformers", "transformers.js"],
+	"zero-shot-object-detection": ["transformers"],
+};
+
+/**
+ * Return the whole TaskData object for a certain task.
+ * If the partialTaskData argument is left undefined,
+ * the default placholder data will be used.
+ */
+function getData(type: PipelineType, partialTaskData: TaskDataCustom = placeholder): TaskData {
+	return {
+		...partialTaskData,
+		id: type,
+		label: PIPELINE_DATA[type].name,
+		libraries: TASKS_MODEL_LIBRARIES[type],
+	};
+}
 
 // To make comparisons easier, task order is the same as in const.ts
 // Tasks set to undefined won't have an associated task page.
@@ -54,6 +128,8 @@ export const TASKS_DATA: Record<PipelineType, TaskData | undefined> = {
 	"image-segmentation": getData("image-segmentation", imageSegmentation),
 	"image-to-image": getData("image-to-image", imageToImage),
 	"image-to-text": getData("image-to-text", imageToText),
+	"image-to-video": undefined,
+	"mask-generation": getData("mask-generation", placeholder),
 	"multiple-choice": undefined,
 	"object-detection": getData("object-detection", objectDetection),
 	"video-classification": getData("video-classification", videoClassification),
@@ -84,18 +160,67 @@ export const TASKS_DATA: Record<PipelineType, TaskData | undefined> = {
 	"voice-activity-detection": undefined,
 	"zero-shot-classification": getData("zero-shot-classification", zeroShotClassification),
 	"zero-shot-image-classification": getData("zero-shot-image-classification", zeroShotImageClassification),
+	"zero-shot-object-detection": getData("zero-shot-object-detection", placeholder),
 } as const;
 
-/**
- * Return the whole TaskData object for a certain task.
- * If the partialTaskData argument is left undefined,
- * the default placholder data will be used.
- */
-function getData(type: PipelineType, partialTaskData: TaskDataCustom = placeholder): TaskData {
-	return {
-		...partialTaskData,
-		id: type,
-		label: PIPELINE_DATA[type].name,
-		libraries: TASKS_MODEL_LIBRARIES[type],
-	};
+export interface ExampleRepo {
+	description: string;
+	id: string;
 }
+
+export type TaskDemoEntry =
+	| {
+			filename: string;
+			type: "audio";
+	  }
+	| {
+			data: Array<{
+				label: string;
+				score: number;
+			}>;
+			type: "chart";
+	  }
+	| {
+			filename: string;
+			type: "img";
+	  }
+	| {
+			table: string[][];
+			type: "tabular";
+	  }
+	| {
+			content: string;
+			label: string;
+			type: "text";
+	  }
+	| {
+			text: string;
+			tokens: Array<{
+				end: number;
+				start: number;
+				type: string;
+			}>;
+			type: "text-with-tokens";
+	  };
+
+export interface TaskDemo {
+	inputs: TaskDemoEntry[];
+	outputs: TaskDemoEntry[];
+}
+
+export interface TaskData {
+	datasets: ExampleRepo[];
+	demo: TaskDemo;
+	id: PipelineType;
+	isPlaceholder?: boolean;
+	label: string;
+	libraries: ModelLibraryKey[];
+	metrics: ExampleRepo[];
+	models: ExampleRepo[];
+	spaces: ExampleRepo[];
+	summary: string;
+	widgetModels: string[];
+	youtubeId?: string;
+}
+
+export type TaskDataCustom = Omit<TaskData, "id" | "label" | "libraries">;

package/src/{object-detection → tasks/object-detection}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{placeholder → tasks/placeholder}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [],

package/src/{question-answering → tasks/question-answering}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{reinforcement-learning → tasks/reinforcement-learning}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{sentence-similarity → tasks/sentence-similarity}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{summarization → tasks/summarization}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{table-question-answering → tasks/table-question-answering}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{tabular-classification → tasks/tabular-classification}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{tabular-regression → tasks/tabular-regression}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{text-classification → tasks/text-classification}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [

package/src/{text-generation → tasks/text-generation}/about.md

@@ -26,11 +26,11 @@ A popular variant of Text Generation models predicts the next word given a bunch
 - Continue a story given the first sentences.
 - Provided a code description, generate the code.
 
-The most popular models for this task are GPT-based models or [Llama series](https://huggingface.co/meta-llama/Llama-2-7b-chat-hf). These models are trained on data that has no labels, so you just need plain text to train your own model. You can train text generation models to generate a wide variety of documents, from code to stories.
+The most popular models for this task are GPT-based models, [Mistral](mistralai/Mistral-7B-v0.1) or [Llama series](https://huggingface.co/meta-llama/Llama-2-7b-chat-hf). These models are trained on data that has no labels, so you just need plain text to train your own model. You can train text generation models to generate a wide variety of documents, from code to stories.
 
 ### Text-to-Text Generation Models
 
-These models are trained to learn the mapping between a pair of texts (e.g. translation from one language to another). The most popular variants of these models are [FLAN-T5](https://huggingface.co/google/flan-t5-xxl), and [BART](https://huggingface.co/docs/transformers/model_doc/bart). Text-to-Text models are trained with multi-tasking capabilities, they can accomplish a wide range of tasks, including summarization, translation, and text classification.
+These models are trained to learn the mapping between a pair of texts (e.g. translation from one language to another). The most popular variants of these models are [NLLB](facebook/nllb-200-distilled-600M), [FLAN-T5](https://huggingface.co/google/flan-t5-xxl), and [BART](https://huggingface.co/docs/transformers/model_doc/bart). Text-to-Text models are trained with multi-tasking capabilities, they can accomplish a wide range of tasks, including summarization, translation, and text classification.
 
 ## Inference
 
@@ -38,7 +38,7 @@ You can use the 🤗 Transformers library `text-generation` pipeline to do infer
 
 ```python
 from transformers import pipeline
-generator = pipeline('text-generation', model = 'gpt2')
+generator = pipeline('text-generation', model = 'HuggingFaceH4/zephyr-7b-beta')
 generator("Hello, I'm a language model", max_length = 30, num_return_sequences=3)
 ## [{'generated_text': "Hello, I'm a language modeler. So while writing this, when I went out to meet my wife or come home she told me that my"},
 ##  {'generated_text': "Hello, I'm a language modeler. I write and maintain software in Python. I love to code, and that includes coding things that require writing"}, ...

package/src/{text-generation → tasks/text-generation}/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "../Types";
+import type { TaskDataCustom } from "..";
 
 const taskData: TaskDataCustom = {
 	datasets: [
@@ -119,7 +119,7 @@ const taskData: TaskDataCustom = {
 	],
 	summary:
 		"Generating text is the task of producing new text. These models can, for example, fill in incomplete text or paraphrase.",
-	widgetModels: ["tiiuae/falcon-7b-instruct"],
+	widgetModels: ["HuggingFaceH4/zephyr-7b-beta"],
 	youtubeId: "Vpjb1lu0MDk",
 };