Tasks: update widget models, model recommendations (#876)

Updated widget models and model recommendations and fixed former canonical model and dataset ids. --------- Co-authored-by: Merve Noyan <[email protected]> Co-authored-by: Pedro Cuenca <[email protected]>
huggingface · Sep 5, 2024 · a087416 · a087416
1 parent 148e413
commit a087416
Show file tree

Hide file tree

Showing 21 changed files with 106 additions and 57 deletions.
diff --git a/packages/tasks/src/tasks/audio-classification/data.ts b/packages/tasks/src/tasks/audio-classification/data.ts
@@ -4,7 +4,11 @@ const taskData: TaskDataCustom = {
 	datasets: [
 		{
 			description: "A benchmark of 10 different audio tasks.",
-			id: "superb",
+			id: "s3prl/superb",
+		},
+		{
+			description: "A dataset of YouTube clips and their sound categories.",
+			id: "agkphysics/AudioSet",
 		},
 	],
 	demo: {
@@ -50,11 +54,11 @@ const taskData: TaskDataCustom = {
 	],
 	models: [
 		{
-			description: "An easy-to-use model for Command Recognition.",
+			description: "An easy-to-use model for command recognition.",
 			id: "speechbrain/google_speech_command_xvector",
 		},
 		{
-			description: "An Emotion Recognition model.",
+			description: "An emotion recognition model.",
 			id: "ehcalabres/wav2vec2-lg-xlsr-en-speech-emotion-recognition",
 		},
 		{
@@ -70,7 +74,7 @@ const taskData: TaskDataCustom = {
 	],
 	summary:
 		"Audio classification is the task of assigning a label or class to a given audio. It can be used for recognizing which command a user is giving or the emotion of a statement, as well as identifying a speaker.",
-	widgetModels: ["facebook/mms-lid-126"],
+	widgetModels: ["MIT/ast-finetuned-audioset-10-10-0.4593"],
 	youtubeId: "KWwzcmG98Ds",
 };
 

diff --git a/packages/tasks/src/tasks/audio-to-audio/data.ts b/packages/tasks/src/tasks/audio-to-audio/data.ts
@@ -44,7 +44,11 @@ const taskData: TaskDataCustom = {
 		},
 		{
 			description: "A speech enhancement model.",
-			id: "speechbrain/metricgan-plus-voicebank",
+			id: "ResembleAI/resemble-enhance",
+		},
+		{
+			description: "A model that can change the voice in a speech recording.",
+			id: "microsoft/speecht5_vc",
 		},
 	],
 	spaces: [

diff --git a/packages/tasks/src/tasks/automatic-speech-recognition/data.ts b/packages/tasks/src/tasks/automatic-speech-recognition/data.ts
@@ -7,8 +7,8 @@ const taskData: TaskDataCustom = {
 			id: "mozilla-foundation/common_voice_17_0",
 		},
 		{
-			description: "An English dataset with 1,000 hours of data.",
-			id: "librispeech_asr",
+			description: "A dataset with 44.6k hours of English speaker data and 6k hours of other language speakers.",
+			id: "parler-tts/mls_eng",
 		},
 		{
 			description: "A multi-lingual audio dataset with 370K hours of audio.",
@@ -54,6 +54,10 @@ const taskData: TaskDataCustom = {
 			description: "An end-to-end model that performs ASR and Speech Translation by MetaAI.",
 			id: "facebook/seamless-m4t-v2-large",
 		},
+		{
+			description: "Powerful speaker diarization model.",
+			id: "pyannote/speaker-diarization-3.1",
+		},
 	],
 	spaces: [
 		{

diff --git a/packages/tasks/src/tasks/document-question-answering/data.ts b/packages/tasks/src/tasks/document-question-answering/data.ts
@@ -46,11 +46,15 @@ const taskData: TaskDataCustom = {
 	],
 	models: [
 		{
-			description: "A LayoutLM model for the document QA task, fine-tuned on DocVQA and SQuAD2.0.",
+			description: "A robust document question answering model.",
 			id: "impira/layoutlm-document-qa",
 		},
 		{
-			description: "A special model for OCR-free Document QA task.",
+			description: "A document question answering model specialized in invoices.",
+			id: "impira/layoutlm-invoices",
+		},
+		{
+			description: "A special model for OCR-free document question answering.",
 			id: "microsoft/udop-large",
 		},
 		{
@@ -74,7 +78,7 @@ const taskData: TaskDataCustom = {
 	],
 	summary:
 		"Document Question Answering (also known as Document Visual Question Answering) is the task of answering questions on document images. Document question answering models take a (document, question) pair as input and return an answer in natural language. Models usually rely on multi-modal features, combining text, position of words (bounding-boxes) and image.",
-	widgetModels: ["impira/layoutlm-document-qa"],
+	widgetModels: ["impira/layoutlm-invoices"],
 	youtubeId: "",
 };
 

diff --git a/packages/tasks/src/tasks/fill-mask/data.ts b/packages/tasks/src/tasks/fill-mask/data.ts
@@ -61,12 +61,12 @@ const taskData: TaskDataCustom = {
 	],
 	models: [
 		{
-			description: "A faster and smaller model than the famous BERT model.",
-			id: "distilbert-base-uncased",
+			description: "The famous BERT model.",
+			id: "google-bert/bert-base-uncased",
 		},
 		{
 			description: "A multilingual model trained on 100 languages.",
-			id: "xlm-roberta-base",
+			id: "FacebookAI/xlm-roberta-base",
 		},
 	],
 	spaces: [],

diff --git a/packages/tasks/src/tasks/image-segmentation/data.ts b/packages/tasks/src/tasks/image-segmentation/data.ts
@@ -92,7 +92,7 @@ const taskData: TaskDataCustom = {
 	],
 	summary:
 		"Image Segmentation divides an image into segments where each pixel in the image is mapped to an object. This task has multiple variants such as instance segmentation, panoptic segmentation and semantic segmentation.",
-	widgetModels: ["facebook/detr-resnet-50-panoptic"],
+	widgetModels: ["nvidia/segformer-b0-finetuned-ade-512-512"],
 	youtubeId: "dKE8SIt9C-w",
 };
 

diff --git a/packages/tasks/src/tasks/image-to-image/data.ts b/packages/tasks/src/tasks/image-to-image/data.ts
@@ -94,7 +94,7 @@ const taskData: TaskDataCustom = {
 	],
 	summary:
 		"Image-to-image is the task of transforming an input image through a variety of possible manipulations and enhancements, such as super-resolution, image inpainting, colorization, and more.",
-	widgetModels: ["lllyasviel/sd-controlnet-canny"],
+	widgetModels: ["stabilityai/stable-diffusion-2-inpainting"],
 	youtubeId: "",
 };
 

diff --git a/packages/tasks/src/tasks/image-to-text/data.ts b/packages/tasks/src/tasks/image-to-text/data.ts
@@ -75,7 +75,7 @@ const taskData: TaskDataCustom = {
 	],
 	summary:
 		"Image to text models output a text from a given image. Image captioning or optical character recognition can be considered as the most common applications of image to text.",
-	widgetModels: ["Salesforce/blip-image-captioning-base"],
+	widgetModels: ["Salesforce/blip-image-captioning-large"],
 	youtubeId: "",
 };
 

diff --git a/packages/tasks/src/tasks/question-answering/data.ts b/packages/tasks/src/tasks/question-answering/data.ts
@@ -52,7 +52,11 @@ const taskData: TaskDataCustom = {
 			id: "deepset/roberta-base-squad2",
 		},
 		{
-			description: "A special model that can answer questions from tables!",
+			description: "Small yet robust model that can answer questions.",
+			id: "distilbert/distilbert-base-cased-distilled-squad",
+		},
+		{
+			description: "A special model that can answer questions from tables.",
 			id: "google/tapas-base-finetuned-wtq",
 		},
 	],

diff --git a/packages/tasks/src/tasks/sentence-similarity/data.ts b/packages/tasks/src/tasks/sentence-similarity/data.ts
@@ -69,8 +69,8 @@ const taskData: TaskDataCustom = {
 			id: "sentence-transformers/all-mpnet-base-v2",
 		},
 		{
-			description: "A multilingual model trained for FAQ retrieval.",
-			id: "clips/mfaq",
+			description: "A multilingual robust sentence similarity model..",
+			id: "BAAI/bge-m3",
 		},
 	],
 	spaces: [
@@ -94,7 +94,7 @@ const taskData: TaskDataCustom = {
 	],
 	summary:
 		"Sentence Similarity is the task of determining how similar two texts are. Sentence similarity models convert input texts into vectors (embeddings) that capture semantic information and calculate how close (similar) they are between them. This task is particularly useful for information retrieval and clustering/grouping.",
-	widgetModels: ["sentence-transformers/all-MiniLM-L6-v2"],
+	widgetModels: ["BAAI/bge-small-en-v1.5"],
 	youtubeId: "VCZq5AkbNEU",
 };
 

diff --git a/packages/tasks/src/tasks/summarization/data.ts b/packages/tasks/src/tasks/summarization/data.ts
@@ -46,7 +46,7 @@ const taskData: TaskDataCustom = {
 		},
 		{
 			description: "A summarization model trained on medical articles.",
-			id: "google/bigbird-pegasus-large-pubmed",
+			id: "Falconsai/medical_summarization",
 		},
 	],
 	spaces: [
@@ -69,7 +69,7 @@ const taskData: TaskDataCustom = {
 	],
 	summary:
 		"Summarization is the task of producing a shorter version of a document while preserving its important information. Some models can extract text from the original input, while other models can generate entirely new text.",
-	widgetModels: ["sshleifer/distilbart-cnn-12-6"],
+	widgetModels: ["facebook/bart-large-cnn"],
 	youtubeId: "yHnr5Dk2zCI",
 };
 

diff --git a/packages/tasks/src/tasks/text-classification/data.ts b/packages/tasks/src/tasks/text-classification/data.ts
@@ -4,11 +4,11 @@ const taskData: TaskDataCustom = {
 	datasets: [
 		{
 			description: "A widely used dataset used to benchmark multiple variants of text classification.",
-			id: "glue",
+			id: "nyu-mll/glue",
 		},
 		{
 			description: "A text classification dataset used to benchmark natural language inference models",
-			id: "snli",
+			id: "stanfordnlp/snli",
 		},
 	],
 	demo: {
@@ -61,11 +61,23 @@ const taskData: TaskDataCustom = {
 	models: [
 		{
 			description: "A robust model trained for sentiment analysis.",
-			id: "distilbert-base-uncased-finetuned-sst-2-english",
+			id: "distilbert/distilbert-base-uncased-finetuned-sst-2-english",
 		},
 		{
-			description: "Multi-genre natural language inference model.",
-			id: "roberta-large-mnli",
+			description: "A sentiment analysis model specialized in financial sentiment.",
+			id: "ProsusAI/finbert",
+		},
+		{
+			description: "A sentiment analysis model specialized in analyzing tweets.",
+			id: "cardiffnlp/twitter-roberta-base-sentiment-latest",
+		},
+		{
+			description: "A model that can classify languages.",
+			id: "papluca/xlm-roberta-base-language-detection",
+		},
+		{
+			description: "A model that can classify text generation attacks.",
+			id: "meta-llama/Prompt-Guard-86M",
 		},
 	],
 	spaces: [
@@ -84,7 +96,7 @@ const taskData: TaskDataCustom = {
 	],
 	summary:
 		"Text Classification is the task of assigning a label or class to a given text. Some use cases are sentiment analysis, natural language inference, and assessing grammatical correctness.",
-	widgetModels: ["distilbert-base-uncased-finetuned-sst-2-english"],
+	widgetModels: ["distilbert/distilbert-base-uncased-finetuned-sst-2-english"],
 	youtubeId: "leNG9fN9FQU",
 };
 

diff --git a/packages/tasks/src/tasks/text-generation/data.ts b/packages/tasks/src/tasks/text-generation/data.ts
@@ -97,8 +97,8 @@ const taskData: TaskDataCustom = {
 			id: "HuggingFaceH4/zephyr-chat",
 		},
 		{
-			description: "An text generation application that combines OpenAI and Hugging Face models.",
-			id: "microsoft/HuggingGPT",
+			description: "A leaderboard that ranks text generation models based on blind votes from people.",
+			id: "lmsys/chatbot-arena-leaderboard",
 		},
 		{
 			description: "An chatbot to converse with a very powerful text generation model.",
@@ -107,7 +107,7 @@ const taskData: TaskDataCustom = {
 	],
 	summary:
 		"Generating text is the task of generating new text given another text. These models can, for example, fill in incomplete text or paraphrase.",
-	widgetModels: ["HuggingFaceH4/zephyr-7b-beta"],
+	widgetModels: ["mistralai/Mistral-Nemo-Instruct-2407"],
 	youtubeId: "e9gNEAlsOvU",
 };
 

diff --git a/packages/tasks/src/tasks/text-to-image/data.ts b/packages/tasks/src/tasks/text-to-image/data.ts
@@ -93,7 +93,7 @@ const taskData: TaskDataCustom = {
 	],
 	summary:
 		"Generates images from input text. These models can be used to generate and modify images based on text prompts.",
-	widgetModels: ["CompVis/stable-diffusion-v1-4"],
+	widgetModels: ["black-forest-labs/FLUX.1-dev"],
 	youtubeId: "",
 };
 

diff --git a/packages/tasks/src/tasks/text-to-speech/data.ts b/packages/tasks/src/tasks/text-to-speech/data.ts
@@ -9,7 +9,7 @@ const taskData: TaskDataCustom = {
 		},
 		{
 			description: "Multi-speaker English dataset.",
-			id: "LibriTTS",
+			id: "mythicinfinity/libritts_r",
 		},
 	],
 	demo: {
@@ -36,11 +36,15 @@ const taskData: TaskDataCustom = {
 	models: [
 		{
 			description: "A powerful TTS model.",
-			id: "suno/bark",
+			id: "parler-tts/parler-tts-large-v1",
 		},
 		{
 			description: "A massively multi-lingual TTS model.",
-			id: "facebook/mms-tts",
+			id: "coqui/XTTS-v2",
+		},
+		{
+			description: "Robust TTS model.",
+			id: "metavoiceio/metavoice-1B-v0.1",
 		},
 		{
 			description: "A prompt based, powerful TTS model.",

diff --git a/packages/tasks/src/tasks/token-classification/data.ts b/packages/tasks/src/tasks/token-classification/data.ts
@@ -4,12 +4,12 @@ const taskData: TaskDataCustom = {
 	datasets: [
 		{
 			description: "A widely used dataset useful to benchmark named entity recognition models.",
-			id: "conll2003",
+			id: "eriktks/conll2003",
 		},
 		{
 			description:
 				"A multilingual dataset of Wikipedia articles annotated for named entity recognition in over 150 different languages.",
-			id: "wikiann",
+			id: "unimelb-nlp/wikiann",
 		},
 	],
 	demo: {
@@ -63,6 +63,14 @@ const taskData: TaskDataCustom = {
 				"A robust performance model to identify people, locations, organizations and names of miscellaneous entities.",
 			id: "dslim/bert-base-NER",
 		},
+		{
+			description: "A strong model to identify people, locations, organizations and names in multiple languages.",
+			id: "FacebookAI/xlm-roberta-large-finetuned-conll03-english",
+		},
+		{
+			description: "A token classification model specialized on medical entity recognition.",
+			id: "blaze999/Medical-NER",
+		},
 		{
 			description: "Flair models are typically the state of the art in named entity recognition tasks.",
 			id: "flair/ner-english",
@@ -77,7 +85,7 @@ const taskData: TaskDataCustom = {
 	],
 	summary:
 		"Token classification is a natural language understanding task in which a label is assigned to some tokens in a text. Some popular token classification subtasks are Named Entity Recognition (NER) and Part-of-Speech (PoS) tagging. NER models could be trained to identify specific entities in a text, such as dates, individuals and places; and PoS tagging would identify, for example, which words in a text are verbs, nouns, and punctuation marks.",
-	widgetModels: ["dslim/bert-base-NER"],
+	widgetModels: ["FacebookAI/xlm-roberta-large-finetuned-conll03-english"],
 	youtubeId: "wVHdVlPScxA",
 };
 

diff --git a/packages/tasks/src/tasks/translation/data.ts b/packages/tasks/src/tasks/translation/data.ts
@@ -5,12 +5,12 @@ const taskData: TaskDataCustom = {
 	datasets: [
 		{
 			description: "A dataset of copyright-free books translated into 16 different languages.",
-			id: "opus_books",
+			id: "Helsinki-NLP/opus_books",
 		},
 		{
 			description:
 				"An example of translation between programming languages. This dataset consists of functions in Java and C#.",
-			id: "code_x_glue_cc_code_to_code_trans",
+			id: "google/code_x_glue_cc_code_to_code_trans",
 		},
 	],
 	demo: {
@@ -42,13 +42,14 @@ const taskData: TaskDataCustom = {
 	],
 	models: [
 		{
-			description: "A model that translates from English to French.",
-			id: "Helsinki-NLP/opus-mt-en-fr",
+			description:
+				"Very powerful model that can translate many languages between each other, especially low-resource languages.",
+			id: "facebook/nllb-200-1.3B",
 		},
 		{
 			description:
 				"A general-purpose Transformer that can be used to translate from English to German, French, or Romanian.",
-			id: "t5-base",
+			id: "google-t5/t5-base",
 		},
 	],
 	spaces: [
@@ -57,12 +58,12 @@ const taskData: TaskDataCustom = {
 			id: "Iker/Translate-100-languages",
 		},
 		{
-			description: "An application that can translate between English, Spanish and Hindi.",
-			id: "EuroPython2022/Translate-with-Bloom",
+			description: "An application that can translate between many languages.",
+			id: "Geonmo/nllb-translation-demo",
 		},
 	],
 	summary: "Translation is the task of converting text from one language to another.",
-	widgetModels: ["t5-small"],
+	widgetModels: ["facebook/mbart-large-50-many-to-many-mmt"],
 	youtubeId: "1JvfrvZgi6c",
 };